optimizedlearning / mechanic
☆36Updated last year
Alternatives and similar repositories for mechanic:
Users that are interested in mechanic are comparing it to the libraries listed below
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 8 months ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆104Updated 3 years ago
- ☆33Updated 2 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 5 months ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆37Updated 4 years ago
- AdaCat☆49Updated 2 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆123Updated last year
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Image augmentation library for Jax☆39Updated last year
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 9 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆171Updated last week
- Parameter-Free Optimizers for Pytorch☆122Updated 11 months ago
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated last year
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- gpu tester detects broken and slow gpus in a cluster☆68Updated 2 years ago
- Differentiable Sorting Networks☆114Updated last year
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Updated 2 years ago
- ☆99Updated 3 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 4 years ago
- Gaussian-Bernoulli Restricted Boltzmann Machines☆104Updated 2 years ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆60Updated last year
- Official code for Long Expressive Memory (ICLR 2022, Spotlight)☆69Updated 3 years ago
- ☆53Updated 8 months ago