facebookresearch / bitsandbytes
Library for 8-bit optimizers and quantization routines.
☆717Updated 2 years ago
Alternatives and similar repositories for bitsandbytes:
Users that are interested in bitsandbytes are comparing it to the libraries listed below
- Fast Block Sparse Matrices for Pytorch☆545Updated 4 years ago
- Prune a model while finetuning or training.☆402Updated 2 years ago
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆609Updated 2 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Accelerate PyTorch models with ONNX Runtime☆358Updated last month
- FastFormers - highly efficient transformer models for NLU☆704Updated last week
- A GPU performance profiling tool for PyTorch models☆505Updated 3 years ago
- Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀☆1,683Updated 5 months ago
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆311Updated last week
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆252Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆263Updated 2 years ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,038Updated 11 months ago
- Code for the ALiBi method for transformer language models (ICLR 2022)☆520Updated last year
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆432Updated 2 years ago
- Understanding the Difficulty of Training Transformers☆328Updated 2 years ago
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,560Updated last year
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Parallelformers: An Efficient Model Parallelization Toolkit for Deployment☆785Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆182Updated 2 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆312Updated last year
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 2 years ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆375Updated last year
- Efficient, check-pointed data loading for deep learning with massive data sets.☆205Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆748Updated last year
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated last year
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆484Updated 3 years ago
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆435Updated 7 months ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago