duyongan/sunstreaker
以jax为后端的类似keras的框架
This project offers a simple, clear framework for quickly building and experimenting with neural network algorithms. It allows researchers and students to input data, define model architectures, and then train and evaluate them to see the results. It's designed for those in deep learning research or education who want to learn algorithms or rapidly test new ideas.
No commits in the last 6 months. Available on PyPI.
Use this if you are a deep learning researcher or student looking for a clear, easy-to-use framework to quickly learn algorithms, experiment with new ideas, or reproduce recent research papers.
Not ideal if you need a framework for deploying large-scale models in a production environment or require fast distributed training capabilities.
Stars
98
Forks
3
Language
Python
License
Apache-2.0
Category
Last pushed
Jan 13, 2023
Commits (30d)
0
Dependencies
7
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/duyongan/sunstreaker"
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/