vopani / jaxtonLinks
100 exercises to learn JAX
☆584Updated 3 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!☆275Updated 3 years ago
- PIX is an image processing library in JAX, for JAX.☆417Updated 3 months ago
- Host repository for the "Reproducible Deep Learning" PhD course☆406Updated 3 years ago
- ☆578Updated last month
- 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…☆738Updated last year
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- MinT: Minimal Transformer Library and Tutorials☆255Updated 2 years ago
- ☆435Updated 8 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- ☆861Updated 3 weeks ago
- CLU lets you write beautiful training loops in JAX.☆346Updated this week
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated last month
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- All about the fundamentals and working of Diffusion Models☆158Updated 2 years ago
- ML Collections is a library of Python Collections designed for ML use cases.☆960Updated last month
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆206Updated last year
- JAX - A curated list of resources https://github.com/google/jax☆1,842Updated 4 months ago
- Source code, datasets and trained models for the paper Learning Advanced Mathematical Computations from Examples (ICLR 2021), by François…☆180Updated 3 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆480Updated 2 years ago
- My repo for training neural nets using pytorch-lightning and hydra☆221Updated 4 months ago
- Lightweight Hyperparameter Optimization 🚂☆147Updated 9 months ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- A library for distributed ML training with PyTorch☆366Updated 2 years ago
- Notebooks for "Probabilistic Machine Learning" book☆203Updated 3 years ago
- A platform for managing machine learning experiments☆858Updated last month
- For optimization algorithm research and development.☆521Updated this week
- Puzzles for exploring transformers☆350Updated 2 years ago
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,065Updated last month
- Turn SymPy expressions into trainable JAX expressions.☆342Updated 2 months ago