Brax

Brax 由 DeepMind 开源，以 JAX 实现高吞吐物理仿真，适合需要海量并行环境与可微分优化的 RL 研究。