vopani / jaxtonLinks
100 exercises to learn JAX
☆579Updated 2 years ago
Alternatives and similar repositories for jaxton
Users that are interested in jaxton are comparing it to the libraries listed below
Sorting:
- All about the fundamental blocks of TF and JAX!☆274Updated 3 years ago
- PIX is an image processing library in JAX, for JAX.☆417Updated 3 months 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…☆732Updated last year
- ☆572Updated 2 weeks ago
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- A platform for managing machine learning experiments☆853Updated 2 weeks ago
- All about the fundamentals and working of Diffusion Models☆155Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆343Updated last month
- ☆855Updated 2 weeks ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 2 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated 3 weeks ago
- Host repository for the "Reproducible Deep Learning" PhD course☆405Updated 3 years ago
- ☆770Updated last year
- MinT: Minimal Transformer Library and Tutorials☆254Updated 2 years ago
- JAX - A curated list of resources https://github.com/google/jax☆1,823Updated 3 months ago
- Notebooks for "Probabilistic Machine Learning" book☆203Updated 3 years ago
- Lightweight Hyperparameter Optimization 🚂☆147Updated 9 months ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆382Updated this week
- Turn SymPy expressions into trainable JAX expressions.☆337Updated last month
- For optimization algorithm research and development.☆518Updated this week
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆204Updated last year