huggingface / accelerate

Related projects ⓘ

Alternatives and complementary repositories for accelerate

• Dao-AILab / flash-attention
  Fast and memory-efficient exact attention
  ☆14,109Updated this week
• huggingface / trl
  Train transformer language models with reinforcement learning.
  ☆9,967Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆5,871Updated 7 months ago
• bitsandbytes-foundation / bitsandbytes
  Accessible large language models via k-bit quantization for PyTorch.
  ☆6,244Updated this week
• huggingface / peft
  🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
  ☆16,354Updated this week
• facebookresearch / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆8,612Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,187Updated 2 months ago
• NVIDIA / Megatron-LM
  Ongoing research training transformer models at scale
  ☆10,480Updated this week
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆4,760Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,500Updated 3 weeks ago
• mlfoundations / open_clip
  An open source implementation of CLIP.
  ☆10,234Updated this week
• huggingface / optimum
  🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
  ☆2,556Updated this week
• microsoft / LoRA
  Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"
  ☆10,661Updated 2 months ago
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
  ☆28,337Updated this week
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,025Updated 6 months ago
• salesforce / LAVIS
  LAVIS - A One-stop Library for Language-Vision Intelligence
  ☆9,898Updated 3 weeks ago
• adapter-hub / adapters
  A Unified Library for Parameter-Efficient and Modular Transfer Learning
  ☆2,566Updated last week
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,392Updated last week
• EleutherAI / lm-evaluation-harness
  A framework for few-shot evaluation of language models.
  ☆6,904Updated this week
• webdataset / webdataset
  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
  ☆2,303Updated last month
• pytorch / serve
  Serve, optimize and scale PyTorch models in production
  ☆4,209Updated 2 weeks ago
• microsoft / unilm
  Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
  ☆20,122Updated last week
• CarperAI / trlx
  A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)
  ☆4,500Updated 10 months ago
• Lightning-AI / torchmetrics
  Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,134Updated this week
• google-research / text-to-text-transfer-transformer
  Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
  ☆6,160Updated last month
• jessevig / bertviz
  BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
  ☆6,919Updated last year
• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆10,252Updated last week
• UKPLab / sentence-transformers
  State-of-the-Art Text Embeddings
  ☆15,236Updated this week
• mlfoundations / open_flamingo
  An open-source framework for training large multimodal models.
  ☆3,732Updated 2 months ago
• huggingface / evaluate
  🤗 Evaluate: A library for easily evaluating machine learning models and datasets.
  ☆2,029Updated last month