greydanus / mnist1d
A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
☆214Updated 5 months ago
Alternatives and similar repositories for mnist1d:
Users that are interested in mnist1d are comparing it to the libraries listed below
- A functional training loops library for JAX☆86Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆169Updated 2 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆369Updated this week
- Compositional Linear Algebra☆464Updated last month
- Parameter-Free Optimizers for Pytorch☆121Updated 10 months ago
- Notebooks for the "JAX in Action" book☆131Updated 8 months ago
- Mathematical operations for JAX pytrees☆198Updated 3 months ago
- The boundary of neural network trainability is fractal☆195Updated last year
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 7 months ago
- ☆212Updated 7 months ago
- CLU lets you write beautiful training loops in JAX.☆335Updated last week
- For optimization algorithm research and development.☆498Updated last week
- Loopy belief propagation for factor graphs on discrete variables in JAX☆142Updated 4 months ago
- 🧱 Modula software package☆169Updated this week
- ☆153Updated 4 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆249Updated last week
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- ☆418Updated 4 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Documentation:☆119Updated last year
- Uncertainty quantification with PyTorch☆344Updated last week
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last month
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- ☆213Updated 3 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Hierarchical Associative Memory User Experience☆99Updated last year
- ☆98Updated 3 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Named Tensors for Legible Deep Learning in JAX☆165Updated this week