JAX
Composable function transforms — the framework Gemini is built with.
Bir yazılım kaydına bakıyorsunuz: yapay zeka yığınındaki bir araç veya platform — kimin geliştirip sürdürdüğü ve hangi donanım üzerinde çalıştığı.
Fotoğraf: Google LLC · Public domain · Wikimedia Commons
JAX treats models as pure functions to be transformed — grad, jit, vmap, pmap — compiling through XLA to TPU pods with exceptional scaling ergonomics. It is Google DeepMind's house framework (Gemini training) and the favorite of much of the research frontier outside PyTorch.
Bu, yığının geri kalanıyla nasıl bağlantılı?
Bu kaydın bağlantılarına hızlı bakış — bir gruba tıklayarak o bölüme atlayın.
Üzerinde çalışır
Sürdüren
Rakipler
İlgili
Kaynaklar
İncelenen kayıt tarihi: 2026-01-15 — görüntüle Metodoloji.