@misc{jaxrl,
  author = {Kostrikov, Ilya},
  doi = {10.5281/zenodo.5535154},
  month = {10},
  title = {{JAXRL: Implementations of Reinforcement Learning algorithms in JAX}},
  url = {https://github.com/ikostrikov/jaxrl2},
  year = {2022},
  note = {v2}
}

pip install --upgrade pip

pip install -e .
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html  # Note: wheels only available on linux.

MUJOCO_GL=egl CUDA_VISIBLE_DEVICES= pytest tests

Contributors

Stargazers

Forkers

egg-west khatch31 moojink stanford-iris-lab joemwatson dohmjan dhruvsreenivas dtbinh oceank vinaykonuru nafyaz