brax

分散式 GPU 加速演化計算框架。包含完整的演化演算法與基準測試問題庫。

模組化強化學習（RL）函式庫（使用 PyTorch、JAX 與 NVIDIA Warp 實作），支援 Gymnasium/Gym、NVIDIA Isaac Lab、MuJoCo Playground 等環境。