jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces. (by ikostrikov)
jaxrl_m
Skeleton for scalable and flexible Jax RL implementations (by dibyaghosh)
jaxrl | jaxrl_m | |
---|---|---|
2 | 1 | |
576 | 49 | |
- | - | |
0.0 | 4.9 | |
over 1 year ago | 10 months ago | |
Jupyter Notebook | Python | |
MIT License | MIT License |
The number of mentions indicates the total number of mentions that we've tracked plus the number of user suggested alternatives.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
jaxrl
Posts with mentions or reviews of jaxrl.
We have used some of these posts to build our list of alternatives
and similar projects. The last one was on 2023-06-20.
-
JAX in Reinforcement Learning
Have you looked at this repo or this repo ?
-
CleanRL now has a DDPG + JAX implementation roughly 2.5-4x faster than DDPG + PyTorch
https://github.com/ikostrikov/jaxrl would be another great reference implementation. Probably you want to also checkout the docs for jax, flax, and optax.
jaxrl_m
Posts with mentions or reviews of jaxrl_m.
We have used some of these posts to build our list of alternatives
and similar projects. The last one was on 2023-06-20.
-
JAX in Reinforcement Learning
Have you looked at this repo or this repo ?
What are some alternatives?
When comparing jaxrl and jaxrl_m you can also consider the following projects:
sbx - SBX: Stable Baselines Jax (SB3 + Jax)
indaba-pracs-2022 - Notebooks for the Practicals at the Deep Learning Indaba 2022.
cleanrl - High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
Popular-RL-Algorithms - PyTorch implementation of Soft Actor-Critic (SAC), Twin Delayed DDPG (TD3), Actor-Critic (AC/A2C), Proximal Policy Optimization (PPO), QT-Opt, PointNet..
uvadlc_notebooks - Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023