HazyResearch / butterfly
Butterfly matrix multiplication in PyTorch
☆166Updated last year
Alternatives and similar repositories for butterfly:
Users that are interested in butterfly are comparing it to the libraries listed below
- Block-sparse primitives for PyTorch☆153Updated 3 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated last week
- ☆201Updated 2 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆183Updated 2 months ago
- Low Precision Arithmetic Simulation in PyTorch☆272Updated 9 months ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- JMP is a Mixed Precision library for JAX.☆191Updated 3 weeks ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆317Updated 2 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- ☆218Updated last week
- A library of GPU kernels for sparse matrix operations.☆257Updated 4 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆379Updated 3 weeks ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- ☆82Updated 5 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆48Updated 3 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Programmable Neural Network Compression☆148Updated 2 years ago
- ☆10Updated 3 years ago
- ☆285Updated last week
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆107Updated 2 years ago
- Implementation for the paper "Latent Weights Do Not Exist: Rethinking Binarized Neural Network Optimization"☆73Updated 5 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆168Updated 2 months ago
- ☆35Updated 2 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 6 months ago
- ☆47Updated 5 years ago