skolai / fewbitLinks
Compression schema for gradients of activations in backward pass
☆44Updated last year
Alternatives and similar repositories for fewbit
Users that are interested in fewbit are comparing it to the libraries listed below
Sorting:
- Learning to Initialize Neural Networks for Stable and Efficient Training☆139Updated 3 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- A library for unit scaling in PyTorch☆125Updated 7 months ago
- Hacks for PyTorch☆19Updated 2 years ago
- ☆112Updated last year
- PyTorch implementation of HashedNets☆36Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- A block oriented training approach for inference time optimization.☆33Updated 10 months ago
- ☆60Updated 5 years ago
- Official implementation of the paper "You Do Not Fully Utilize Transformer's Representation Capacity"☆29Updated last month
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆117Updated 3 years ago
- ☆29Updated 2 years ago
- Latest Weight Averaging (NeurIPS HITY 2022)☆30Updated 2 years ago
- ☆51Updated last year
- ☆70Updated 10 months ago
- ☆74Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- ☆36Updated last year
- sigma-MoE layer☆20Updated last year
- DropIT: Dropping Intermediate Tensors for Memory-Efficient DNN Training (ICLR 2023)☆31Updated 2 years ago
- The official implementation of the ChordMixer architecture.☆61Updated 2 years ago
- ☆36Updated 7 months ago
- ☆53Updated 9 months ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- ☆157Updated last year
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated 3 weeks ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago
- Identify a binary weight or binary weight and activation subnetwork within a randomly initialized network by only pruning and binarizing …☆52Updated 3 years ago