HazyResearch / butterfly
Butterfly matrix multiplication in PyTorch
☆167Updated 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
- ☆202Updated 2 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated this week
- Customized matrix multiplication kernels☆53Updated 3 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Structured matrices for compressing neural networks☆66Updated last year
- Low Precision Arithmetic Simulation in PyTorch☆272Updated 9 months ago
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- ☆288Updated last week
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆76Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆169Updated 2 months ago
- ☆35Updated 3 months ago
- ☆220Updated last month
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 7 months ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆130Updated 5 years ago
- Programmable Neural Network Compression☆148Updated 2 years ago
- ☆143Updated last year
- ☆10Updated 3 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆382Updated this week
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- A library for unit scaling in PyTorch☆124Updated 3 months ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆48Updated 4 years ago
- ☆161Updated 8 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch☆108Updated last year
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆50Updated 7 years ago