vopani / jaxton
100 exercises to learn JAX
☆576Updated 2 years ago
Alternatives and similar repositories for jaxton:
Users that are interested in jaxton are comparing it to the libraries listed below
- All about the fundamental blocks of TF and JAX!☆276Updated 3 years ago
- 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 o…☆714Updated last year
- A High Level API for Deep Learning in JAX☆473Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆412Updated 3 weeks ago
- ☆841Updated 2 weeks ago
- ☆423Updated 5 months ago
- ☆559Updated this week
- JAX - A curated list of resources https://github.com/google/jax☆1,747Updated last month
- CLU lets you write beautiful training loops in JAX.☆335Updated 3 weeks ago
- Host repository for the "Reproducible Deep Learning" PhD course☆406Updated 2 years ago
- For optimization algorithm research and development.☆502Updated this week
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆891Updated last month
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- A platform for managing machine learning experiments☆841Updated 2 weeks ago
- Puzzles for exploring transformers☆335Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last month
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Turn SymPy expressions into trainable JAX expressions.☆332Updated 2 months ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,285Updated last week
- All about the fundamentals and working of Diffusion Models☆155Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆154Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆956Updated 2 weeks ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,421Updated 8 months ago
- ☆771Updated last year
- Library for reading and processing ML training data.☆414Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆251Updated this week
- MinT: Minimal Transformer Library and Tutorials☆253Updated 2 years ago
- Notebooks for the Practicals at the Deep Learning Indaba 2022.☆176Updated last year