🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
Primary LanguagePythonApache License 2.0Apache-2.0