JAX
Software
Entrenamiento
Agentes
Composable function transforms — the framework Gemini is built with.
Estás viendo un registro de software: una herramienta o plataforma en el stack de IA, quién la mantiene y en qué hardware se ejecuta.
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.
Cómo se conecta con el resto del stack
Los enlaces de esta ficha de un vistazo: haz clic en un grupo para ir directamente a él.
Se ejecuta en
Mantenido por
Compite con
Relacionado
Fuentes
Registro revisado a fecha de 2026-01-15 — ver Metodología.