greydanus / mnist1d
A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
☆220Updated 6 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…☆173Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆375Updated last week
- Parameter-Free Optimizers for Pytorch☆123Updated last year
- Compositional Linear Algebra☆474Updated 3 weeks ago
- Mathematical operations for JAX pytrees☆199Updated 4 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆115Updated 3 weeks ago
- Uncertainty quantification with PyTorch☆348Updated last week
- 🧱 Modula software package☆188Updated 3 weeks ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆254Updated last week
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 4 months ago
- A functional training loops library for JAX☆86Updated last year
- Riemannian Optimization Using JAX☆48Updated last year
- Running Jax in PyTorch Lightning☆94Updated 4 months ago
- Neural Networks for JAX☆84Updated 7 months ago
- Code for our NeurIPS 2022 paper☆367Updated 2 years ago
- Named Tensors for Legible Deep Learning in JAX☆172Updated last week
- A framework for conducting machine learning experiments in python☆40Updated 4 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 9 months ago
- ☆216Updated 9 months ago
- Lightweight Hyperparameter Optimization 🚂☆147Updated 7 months ago
- Hierarchical Associative Memory User Experience☆101Updated last year
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆578Updated 3 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆36Updated last week
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- ☆60Updated 3 years ago
- ☆99Updated 3 years ago
- ☆155Updated last year
- Multiple dispatch over abstract array types in JAX.☆117Updated this week
- IVON optimizer for neural networks based on variational learning.☆62Updated 5 months ago
- Laplace approximations for Deep Learning.☆502Updated this week