greydanus / mnist1dLinks
A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
☆225Updated 9 months ago
Alternatives and similar repositories for mnist1d
Users that are interested in mnist1d are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated last month
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated 2 months ago
- Uncertainty quantification with PyTorch☆362Updated 3 months ago
- Mathematical operations for JAX pytrees☆198Updated 7 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆188Updated 7 months ago
- Deep Learning, an Energy Approach☆192Updated last month
- Compositional Linear Algebra☆478Updated last month
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆243Updated 4 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆270Updated last week
- Hierarchical Associative Memory User Experience☆101Updated this week
- ☆60Updated 3 years ago
- ☆247Updated 2 weeks ago
- A functional training loops library for JAX☆88Updated last year
- 🧱 Modula software package☆207Updated 3 months ago
- ☆157Updated last year
- Agustinus' very opiniated publication-ready plotting library☆67Updated 2 months ago
- ☆214Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated 11 months ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Riemannian Optimization Using JAX☆49Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆388Updated last week
- Stochastic Automatic Differentiation library for PyTorch.☆204Updated 10 months ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- ☆100Updated 3 years ago
- Materials of the Nordic Probabilistic AI School 2022.☆181Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆130Updated last month
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago