开发者生态
JAX
JAX 由 Google 等团队维护，以可组合变换（grad、jit、vmap）著称，是机器人学习（尤其 JAX 系仿真与算法）的底层引擎。
项目说明
JAX 是带自动微分与 JIT 的 NumPy 风格数值计算框架。
它支撑 Brax 等仿真与大量 RL 研究，是机器人学习的重要底层工具。
能力与适用场景
任务域
rlimitationvla
框架 / 工具链
JAXFlaxOptax
主题标签
autodiffnumpymachine-learning