🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
instadeepai
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
EdanToledo
