search

mansicer / baselax

Baselax (Baselines + JAX) provides stable-baselines-style implementations of reinforcement learning (RL) algorithms with Google JAX framework.
https://baselax.readthedocs.io
Apache License 2.0
4 stars 1 forks source link

readme

Status: in development, no practical use.

Baselax

Baselax (Baselines + jax) provides stable-baselines-style implementations of reinforcement learning (RL) algorithms with Google JAX framework, supported by the following frameworks:

Installation

Install via pip:

pip install -r requirements.txt
pip install git+https://github.com/deepmind/dm-haiku

Additnionally, you can download and install envpool wheels from GitHub releases for better sample efficiency.