Brax 由 DeepMind 开源，以 JAX 实现高吞吐物理仿真，适合需要海量并行环境与可微分优化的 RL 研究。
Brax 是基于 JAX 的物理仿真器，支持大规模并行与可微分计算。
它可在加速硬件上同时训练大量环境，适合强化学习策略研究。