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.
☆749Updated 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,904Updated 6 months ago
- 100 exercises to learn JAX☆589Updated 3 years ago
- ☆875Updated 3 weeks ago
- All about the fundamental blocks of TF and JAX!☆276Updated 3 years ago
- Optax is a gradient processing and optimization library for JAX.☆1,980Updated last week
- CLU lets you write beautiful training loops in JAX.☆355Updated last month
- ☆584Updated this week
- ☆783Updated 2 months ago
- PIX is an image processing library in JAX, for JAX.☆421Updated 5 months ago
- ML Collections is a library of Python Collections designed for ML use cases.☆970Updated last month
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆990Updated 4 months ago
- Library for reading and processing ML training data.☆505Updated this week
- Course notes☆708Updated last year
- ☆444Updated 10 months ago
- ☆812Updated 4 months ago
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆174Updated last year
- A Graph Neural Network Library in Jax☆1,444Updated last year
- "Deep Generative Modeling": Introductory Examples☆1,224Updated last week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,472Updated last week
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.☆610Updated 2 weeks ago
- JAX-based neural network library☆3,078Updated 2 weeks ago
- A walkthrough of transformer architecture code☆352Updated last year
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆213Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆389Updated this week
- TensorDict is a pytorch dedicated tensor container.☆955Updated this week
- Compositional Linear Algebra☆489Updated 3 weeks ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- For optimization algorithm research and development.☆530Updated this week
- Materials of the Nordic Probabilistic AI School 2022.☆181Updated 2 years ago
- All about the fundamentals and working of Diffusion Models☆159Updated 2 years ago