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.
☆758Updated 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,924Updated last month
- 100 exercises to learn JAX☆591Updated 3 years ago
- All about the fundamental blocks of TF and JAX!☆276Updated 3 years ago
- ☆786Updated last month
- Course notes☆721Updated last year
- ☆882Updated last month
- ☆815Updated 5 months ago
- Optax is a gradient processing and optimization library for JAX.☆2,022Updated last week
- All about the fundamentals and working of Diffusion Models☆159Updated 2 years ago
- ML Collections is a library of Python Collections designed for ML use cases.☆982Updated last month
- PIX is an image processing library in JAX, for JAX.☆422Updated 6 months ago
- Tensors, for human consumption☆1,291Updated 3 months ago
- For optimization algorithm research and development.☆539Updated this week
- A walkthrough of transformer architecture code☆359Updated last year
- ☆455Updated 11 months ago
- CLU lets you write beautiful training loops in JAX.☆356Updated 3 months ago
- ☆593Updated last month
- TensorDict is a pytorch dedicated tensor container.☆967Updated this week
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,088Updated last week
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,532Updated this week
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Implementation of https://srush.github.io/annotated-s4☆502Updated 3 months ago
- Library for reading and processing ML training data.☆548Updated this week
- "Deep Generative Modeling": Introductory Examples☆1,250Updated last month
- Materials of the Nordic Probabilistic AI School 2022.☆179Updated 3 years ago
- Notebooks for the "Deep Learning with JAX" book☆155Updated 3 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆996Updated 5 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆398Updated last week
- Everything you want to know about Google Cloud TPU☆546Updated last year