ethanluoyc/corax
Corax: Core RL in JAX
This is for reinforcement learning researchers who need to develop and test new machine learning agents using the JAX framework. It provides modular components and pre-built, high-quality agents that can be customized. Researchers input existing datasets or create new ones, and the output is a trained reinforcement learning agent ready for evaluation or further research.
No commits in the last 6 months.
Use this if you are a reinforcement learning researcher developing new algorithms or building upon existing ones, specifically within the JAX ecosystem.
Not ideal if you are looking for a plug-and-play solution for applying reinforcement learning without deep technical involvement in algorithm development.
Stars
41
Forks
—
Language
Python
License
Apache-2.0
Category
Last pushed
Feb 22, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ethanluoyc/corax"
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/