Alternatives and similar repositories for Megatron-LM:

Users that are interested in Megatron-LM are comparing it to the libraries listed below

• Dao-AILab / flash-attention
  Fast and memory-efficient exact attention
  ☆15,541Updated this week
• huggingface / accelerate
  🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…
  ☆8,343Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆6,025Updated 10 months ago
• bitsandbytes-foundation / bitsandbytes
  Accessible large language models via k-bit quantization for PyTorch.
  ☆6,679Updated this week
• huggingface / peft
  🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
  ☆17,363Updated this week
• facebookresearch / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆9,063Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,260Updated last month
• huggingface / trl
  Train transformer language models with reinforcement learning.
  ☆11,782Updated this week
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆14,452Updated this week
• artidoro / qlora
  QLoRA: Efficient Finetuning of Quantized LLMs
  ☆10,246Updated 8 months ago
• jessevig / bertviz
  BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
  ☆7,130Updated last year
• deepspeedai / DeepSpeed
  DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
  ☆36,800Updated this week
• microsoft / LoRA
  Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"
  ☆11,304Updated 2 months ago
• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆10,598Updated last week
• facebookresearch / metaseq
  Repo for external large-scale work
  ☆6,517Updated 9 months ago
• microsoft / unilm
  Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
  ☆20,762Updated last week
• deepspeedai / Megatron-DeepSpeed
  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  ☆1,987Updated last week
• EleutherAI / lm-evaluation-harness
  A framework for few-shot evaluation of language models.
  ☆7,848Updated this week
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,537Updated 2 weeks ago
• NVIDIA / TensorRT-LLM
  TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain…
  ☆9,441Updated this week
• NVIDIA / NeMo
  A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Auto…
  ☆13,150Updated this week
• vllm-project / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆38,475Updated this week
• harvardnlp / annotated-transformer
  An annotated implementation of the Transformer paper.
  ☆5,993Updated 10 months ago
• karpathy / minGPT
  A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
  ☆21,365Updated 6 months ago
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,046Updated 10 months ago
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆5,066Updated this week
• mit-han-lab / streaming-llm
  [ICLR 2024] Efficient Streaming Language Models with Attention Sinks
  ☆6,795Updated 7 months ago
• sgl-project / sglang
  SGLang is a fast serving framework for large language models and vision language models.
  ☆10,095Updated this week
• wandb / wandb
  The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
  ☆9,505Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,741Updated last week