JAX
Software
Treinamento
Ativo
Composable function transforms — the framework Gemini is built with.
Está a ver um registo de software: uma ferramenta ou plataforma na stack de IA — quem a mantém e em que hardware corre.
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.
Como este elemento se liga ao resto da stack
As ligações deste registo num relance — clique num grupo para ir diretamente até ele.
Executa em
Mantido por
Concorre com
Relacionados
Fontes
Registro revisado em 2026-01-15 — ver Metodologia.