facebookresearch / bitsandbytes
Library for 8-bit optimizers and quantization routines.
☆718Updated 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☆546Updated 4 years ago
- Prune a model while finetuning or training.☆399Updated 2 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- FastFormers - highly efficient transformer models for NLU☆704Updated last year
- Implementation of a Transformer, but completely in Triton☆259Updated 2 years ago
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆612Updated 2 years ago
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆434Updated 6 months ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,032Updated 10 months 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,557Updated last year
- A GPU performance profiling tool for PyTorch models☆501Updated 3 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated last year
- Accelerate PyTorch models with ONNX Runtime☆358Updated last week
- Code for the ALiBi method for transformer language models (ICLR 2022)☆516Updated last year
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆180Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆483Updated 3 years ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆370Updated last year
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆309Updated last week
- Implementation of https://arxiv.org/abs/1904.00962☆371Updated 4 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- ☆338Updated 10 months ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆205Updated last year
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,412Updated this week
- Understanding the Difficulty of Training Transformers☆328Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆374Updated 11 months ago
- Ranger deep learning optimizer rewrite to use newest components☆328Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆747Updated last year
- Pytorch library for fast transformer implementations☆1,682Updated last year