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.
☆746Updated 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,879Updated 5 months ago
- 100 exercises to learn JAX☆586Updated 3 years ago
- All about the fundamental blocks of TF and JAX!☆275Updated 3 years ago
- ☆812Updated 3 months ago
- CLU lets you write beautiful training loops in JAX.☆350Updated last month
- ☆869Updated last week
- Optax is a gradient processing and optimization library for JAX.☆1,959Updated this week
- ML Collections is a library of Python Collections designed for ML use cases.☆968Updated 2 weeks ago
- ☆581Updated 3 weeks ago
- Course notes☆704Updated last year
- Library for reading and processing ML training data.☆484Updated last week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,439Updated this week
- ☆781Updated last month
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆983Updated 3 months ago
- TensorDict is a pytorch dedicated tensor container.☆945Updated this week
- PIX is an image processing library in JAX, for JAX.☆420Updated 4 months ago
- ☆443Updated 9 months ago
- "Deep Generative Modeling": Introductory Examples☆1,208Updated 10 months ago
- For optimization algorithm research and development.☆525Updated this week
- Materials of the Nordic Probabilistic AI School 2022.☆181Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆170Updated last year
- A Graph Neural Network Library in Jax☆1,436Updated last year
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.☆606Updated 3 weeks ago
- A walkthrough of transformer architecture code☆351Updated last year
- Everything you want to know about Google Cloud TPU☆536Updated last year
- Orbax provides common checkpointing and persistence utilities for JAX users☆410Updated this week
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Updated this week
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆254Updated 4 months ago