Beranda › Perangkat Lunak › JAX
JAX
Perangkat Lunak
Pelatihan
Agen
Composable function transforms — the framework Gemini is built with.
Anda sedang melihat entri perangkat lunak: sebuah alat atau platform dalam AI stack — siapa yang memeliharanya dan perangkat keras apa yang dijalankannya.
Foto: 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.
Bagaimana ini terhubung ke bagian lain dari stack
Tautan catatan ini sekilas — klik grup untuk langsung menuju ke sana.
Berjalan di
Dipelihara oleh
Bersaing dengan
Terkait
Sumber
Rekaman ditinjau per 2026-01-15 — lihat Metodologi.