hyeon95y / SparseLinear
A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently
☆48Updated 9 months ago
Related projects: ⓘ
- Structured matrices for compressing neural networks☆65Updated 11 months ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆97Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 3 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆66Updated last year
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆68Updated 2 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆105Updated this week
- Pytorch library for factorized L0-based pruning.☆42Updated 11 months ago
- Distributed K-FAC Preconditioner for PyTorch☆75Updated last week
- Sequence Modeling with Structured State Spaces☆60Updated 2 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 2 years ago
- ☆49Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆103Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆93Updated 4 years ago
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆57Updated 3 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆47Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- A centralized place for deep thinking code and experiments☆73Updated last year
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆34Updated last year
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆45Updated 3 years ago
- PyTorch implementation of HashedNets☆35Updated last year
- ☆96Updated 2 years ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆57Updated 10 months ago
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆41Updated 8 months ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆39Updated 4 years ago
- Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)☆58Updated 2 years ago
- ☆42Updated 3 months ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆77Updated 2 years ago
- Implementation of deep implicit attention in PyTorch☆63Updated 3 years ago