/Stoix

🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers