greydanus / mnist1d
A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
☆213Updated 4 months ago
Alternatives and similar repositories for mnist1d:
Users that are interested in mnist1d are comparing it to the libraries listed below
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆168Updated 2 months ago
- The boundary of neural network trainability is fractal☆194Updated last year
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆241Updated 4 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆364Updated this week
- Parameter-Free Optimizers for Pytorch☆113Updated 9 months ago
- ☆416Updated 4 months ago
- A functional training loops library for JAX☆86Updated last year
- Compositional Linear Algebra☆462Updated 2 weeks ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- 🧱 Modula software package☆146Updated this week
- Uncertainty quantification with PyTorch☆338Updated this week
- ☆59Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- Mathematical operations for JAX pytrees☆198Updated 2 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆111Updated this week
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆103Updated last year
- Image augmentation library for Jax☆37Updated 10 months ago
- Notebooks for the "JAX in Action" book☆130Updated 8 months ago
- ☆153Updated 4 years ago
- ☆211Updated 7 months ago
- Hessian spectral density estimation in TF and Jax☆121Updated 4 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆572Updated last month
- Named Tensors for Legible Deep Learning in JAX☆161Updated last week
- A framework for conducting machine learning experiments in python☆38Updated 2 months ago
- JAX Arrays for human consumption☆90Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- Laplace approximations for Deep Learning.☆496Updated 2 weeks ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆211Updated this week
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆270Updated last year