JAX
Software
Training
Actief
Composable function transforms — the framework Gemini is built with.
U bekijkt een softwarerecord: een tool of platform in de AI-stack — wie het beheert en op welke hardware het draait.
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.
Hoe dit aansluit op de rest van de stack
De koppelingen van dit record in één oogopslag — klik op een groep om er naartoe te springen.
Draait op
Onderhouden door
Concurreert met
Gerelateerd
Bronnen
Record beoordeeld per 2026-01-15 — bekijk Methodologie.