facebookexperimental / protoquantLinks
Prototype routines for GPU quantization written using PyTorch.
☆21Updated 5 months ago
Alternatives and similar repositories for protoquant
Users that are interested in protoquant are comparing it to the libraries listed below
Sorting:
- ☆21Updated 4 months ago
- Torch Distributed Experimental☆116Updated 11 months ago
- Memory Optimizations for Deep Learning (ICML 2023)☆64Updated last year
- Experiment of using Tangent to autodiff triton☆79Updated last year
- PyTorch centric eager mode debugger☆47Updated 7 months ago
- Hacks for PyTorch☆19Updated 2 years ago
- ☆157Updated last year
- ☆28Updated 6 months ago
- A block oriented training approach for inference time optimization.☆33Updated 10 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated 3 weeks ago
- ☆73Updated 3 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 11 months ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Repository for CPU Kernel Generation for LLM Inference☆26Updated 2 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆158Updated 3 weeks ago
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆70Updated last year
- A bunch of kernels that might make stuff slower 😉☆54Updated last week
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- TORCH_LOGS parser for PT2☆46Updated 2 weeks ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Benchmarks to capture important workloads.☆31Updated 5 months ago
- python package of rocm-smi-lib☆21Updated 9 months ago
- extensible collectives library in triton☆87Updated 3 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆42Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆80Updated 10 months ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago
- A library for unit scaling in PyTorch☆125Updated 7 months ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 9 months ago
- PyTorch RFCs (experimental)☆133Updated last month
- ☆36Updated 7 months ago