facebookresearch / bitsandbytes

Related projects: ⓘ

• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆546Updated 3 years ago
• huggingface / nn_pruning
  Prune a model while finetuning or training.
  ☆392Updated 2 years ago
• Lightning-Universe / lightning-transformers
  Flexible components pairing 🤗 Transformers with Pytorch Lightning
  ☆610Updated last year
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆711Updated 9 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆994Updated 5 months ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆242Updated 2 years ago
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆419Updated 3 weeks ago
• pytorch / ort
  Accelerate PyTorch models with ONNX Runtime
  ☆353Updated 2 weeks ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆493Updated 3 years ago
• ofirpress / attention_with_linear_biases
  Code for the ALiBi method for transformer language models (ICLR 2022)
  ☆497Updated 10 months ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,013Updated last month
• microsoft / fastformers
  FastFormers - highly efficient transformer models for NLU
  ☆700Updated 8 months ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆295Updated 3 years ago
• AminRezaei0x443 / memory-efficient-attention
  Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
  ☆178Updated last year
• cybertronai / pytorch-lamb
  Implementation of https://arxiv.org/abs/1904.00962
  ☆366Updated 3 years ago
• tunib-ai / parallelformers
  Parallelformers: An Efficient Model Parallelization Toolkit for Deployment
  ☆776Updated last year
• ELS-RD / kernl
  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,519Updated 7 months ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,334Updated 2 months ago
• lucidrains / memory-efficient-attention-pytorch
  Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"
  ☆355Updated last year
• luyug / GradCache
  Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
  ☆342Updated 5 months ago
• facebookresearch / recipes
  Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…
  ☆286Updated this week
• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆708Updated 3 weeks ago
• google / flaxformer
  ☆322Updated 5 months ago
• microsoft / tutel
  Tutel MoE: An Optimized Mixture-of-Experts Implementation
  ☆711Updated this week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,389Updated this week
• microsoft / fastseq
  An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…
  ☆432Updated 2 years ago
• LiyuanLucasLiu / Transformer-Clinic
  Understanding the Difficulty of Training Transformers
  ☆325Updated 2 years ago
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,116Updated this week
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆356Updated 6 years ago