Baselax (Baselines + JAX) provides stable-baselines-style implementations of reinforcement learning (RL) algorithms with Google JAX framework.
Primary LanguagePythonApache License 2.0Apache-2.0
No issues in this repository yet.