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
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆104Updated 3 years ago
- Easy Hypernetworks in Pytorch and Jax☆98Updated 2 years ago
- ☆33Updated 2 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 4 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Gaussian-Bernoulli Restricted Boltzmann Machines☆102Updated 2 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 7 months ago
- ☆17Updated 9 months ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 years ago
- Parameter-Free Optimizers for Pytorch☆122Updated 10 months ago
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆169Updated 3 months ago
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- AdaCat☆49Updated 2 years ago
- ☆52Updated 5 months ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆99Updated 2 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆37Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆59Updated last year
- ☆98Updated 3 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Efficient Householder Transformation in PyTorch☆64Updated 3 years ago
- ☆47Updated 2 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- 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
- Image augmentation library for Jax☆38Updated 11 months ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆140Updated 3 years ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆15Updated 2 years ago