🧠 AI知识库

简介

Google开源的函数式数值计算框架，支持自动微分和JIT编译到XLA后端。以纯函数、向量化变换（vmap/pmap）和函数式风格著称。DeepMind大量使用JAX进行前沿研究，Google Gemini的训练也基于JAX，在研究圈有极高影响力。