{"data":{"slug":"jax","etype":"software","name":"JAX","cat":"Training","subtitle":"Composable function transforms — the framework Gemini is built with.","summary":"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.","attrs":null,"kf":{"maker":"Google","since":"2018","license":"Apache 2.0"},"aliases":["Flax"],"ticker":null,"exchange":null,"hq":null,"founded":null,"website":null,"status":"Active","region":null,"asof":"2026-01-15","sources":[{"url":"https://github.com/jax-ml/jax","label":"JAX"}]},"meta":{"found":true,"count":1,"served_from":"site cache (no live upstream calls)","as_of":"2026-10-14T09:46:22+00:00","disclaimer":"Research and information only — not investment advice. Structured records carry basis labels (vendor spec / reported / estimate); market quotes are delayed/indicative.","attribution":"Records compiled from vendor documentation, official filings and reputable public reporting; market quotes via financial data providers. See /data-sources."}}