CRAX: Fast Safe Reinforcement Learning Benchmarking
CRAX is a hardware-accelerated benchmark using MuJoCo XLA to test Safe Reinforcement Learning agents efficiently. It provides up to 100x speedups over CPU benchmarks, allowing large-scale experimentation across six environment suites and three agent types. The work reveals crucial trade-offs between performance and safety when training RL policies under cost constraints.