poets-ai/elegy
A High Level API for Deep Learning in JAX
This project helps machine learning practitioners build and train deep learning models efficiently. It takes raw data, applies a deep learning architecture, and outputs trained models ready for making predictions. Data scientists, machine learning engineers, and researchers can use this to quickly prototype and deploy models.
476 stars. No commits in the last 6 months. Available on PyPI.
Use this if you are developing deep learning models with JAX and want a high-level, Keras-like API for common tasks or a flexible low-level API for custom model implementations.
Not ideal if you prefer a different deep learning framework like TensorFlow or PyTorch and are not already working within the JAX ecosystem.
Stars
476
Forks
31
Language
Python
License
MIT
Category
Last pushed
Dec 15, 2022
Commits (30d)
0
Dependencies
4
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/poets-ai/elegy"
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/