yardenas/jax-dreamer
Dreamer on JAX
This project helps machine learning researchers who are working on reinforcement learning. It allows them to experiment with and apply the Dreamer algorithm, a model-based reinforcement learning approach, using JAX. Researchers can input their environment configurations and receive trained policies for decision-making.
No commits in the last 6 months.
Use this if you are a machine learning researcher or practitioner specifically looking to implement or experiment with the Dreamer algorithm for model-based reinforcement learning.
Not ideal if you are new to reinforcement learning or not specifically interested in the Dreamer algorithm, as it requires a foundational understanding of the field.
Stars
16
Forks
1
Language
Python
License
MIT
Category
Last pushed
Jan 19, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/yardenas/jax-dreamer"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/