gordicaleksa / get-started-with-JAXLinks
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
☆760Updated last year
Alternatives and similar repositories for get-started-with-JAX
Users that are interested in get-started-with-JAX are comparing it to the libraries listed below
Sorting:
- JAX - A curated list of resources https://github.com/google/jax☆1,944Updated last month
- 100 exercises to learn JAX☆592Updated 3 years ago
- ☆885Updated last month
- Optax is a gradient processing and optimization library for JAX.☆2,044Updated last week
- All about the fundamental blocks of TF and JAX!☆276Updated 3 years ago
- CLU lets you write beautiful training loops in JAX.☆356Updated 3 months ago
- ☆821Updated 6 months ago
- ☆785Updated last month
- PIX is an image processing library in JAX, for JAX.☆423Updated 7 months ago
- ☆599Updated last week
- ML Collections is a library of Python Collections designed for ML use cases.☆987Updated last month
- "Deep Generative Modeling": Introductory Examples☆1,255Updated last month
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,613Updated this week
- A Graph Neural Network Library in Jax☆1,450Updated last year
- TensorDict is a pytorch dedicated tensor container.☆972Updated this week
- ☆1,363Updated 3 weeks ago
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆186Updated 2 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆999Updated 2 weeks ago
- ☆456Updated last year
- Library for reading and processing ML training data.☆570Updated this week
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆220Updated last year
- Tensors, for human consumption☆1,316Updated 2 weeks ago
- All about the fundamentals and working of Diffusion Models☆159Updated 2 years ago
- Evolution Strategies in JAX 🦎☆672Updated last month
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Updated 2 months ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,579Updated 3 weeks ago
- For optimization algorithm research and development.☆542Updated last week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,094Updated last week
- JAX-based neural network library☆3,107Updated 3 weeks ago
- Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023☆2,993Updated 7 months ago