GitHub repository: BepfCp/baselax
Primary language: Python
Description: Baselax (Baselines + JAX) provides stable-baselines-style implementations of reinforcement learning (RL) algorithms with Google JAX framework.