aredden / torch-bnb-fp4Links
Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops
☆30Updated last year
Alternatives and similar repositories for torch-bnb-fp4
Users that are interested in torch-bnb-fp4 are comparing it to the libraries listed below
Sorting:
- ☆160Updated 2 years ago
- ☆124Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆42Updated last year
- A library for unit scaling in PyTorch☆133Updated 5 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆90Updated last year
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆73Updated last year
- ☆30Updated last year
- Low-bit optimizers for PyTorch☆137Updated 2 years ago
- A block oriented training approach for inference time optimization.☆34Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆79Updated last year
- ☆157Updated 10 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆74Updated last year
- Official implementation for Training LLMs with MXFP4☆116Updated 8 months ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- ☆115Updated last year
- This repository contains code for the MicroAdam paper.☆21Updated last year
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆146Updated last year
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- ☆150Updated 2 years ago
- QuIP quantization☆61Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆108Updated 2 months ago
- Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees" adapted for Llama models☆41Updated 2 years ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆93Updated 11 months ago
- ☆84Updated 11 months ago
- research impl of Native Sparse Attention (2502.11089)☆63Updated 10 months ago
- ☆204Updated last year
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆225Updated 6 months ago
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆118Updated last year