Alternatives and similar repositories for torchforge

Users that are interested in torchforge are comparing it to the libraries listed below

• NVIDIA-NeMo / RL
  Scalable toolkit for efficient model reinforcement
  ☆1,048Updated this week
• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆306Updated last week
• thinking-machines-lab / batch_invariant_ops
  ☆912Updated 3 weeks ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆305Updated 3 weeks ago
• radixark / miles
  ☆317Updated this week
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆848Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆335Updated this week
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆455Updated 2 weeks ago
• microsoft / dion
  Dion optimizer algorithm
  ☆388Updated last week
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆302Updated 3 weeks ago
• huggingface / picotron_tutorial
  ☆225Updated last month
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆780Updated last week
• allenai / OLMo-core
  PyTorch building blocks for the OLMo ecosystem
  ☆400Updated this week
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆271Updated this week
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆254Updated this week
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆629Updated this week
• NVIDIA-NeMo / Automodel
  Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
  ☆187Updated this week
• deepseek-ai / LPLB
  An early research stage MoE load balancer based on inear programming.
  ☆415Updated last week
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆358Updated 11 months ago
• meta-pytorch / monarch
  PyTorch Single Controller
  ☆905Updated this week
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆163Updated last week
• apple / ml-recurrent-drafter
  ☆218Updated 10 months ago
• thinking-machines-lab / tinker
  Training API and CLI
  ☆238Updated this week
• tilde-research / MoMoE-impl
  Memory optimized Mixture of Experts
  ☆69Updated 4 months ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆345Updated this week
• pyember / ember
  ☆233Updated 5 months ago
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models
  ☆209Updated this week
• mingyin0312 / RLFromScratch
  ☆459Updated 3 months ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆160Updated last year
• facebookresearch / PhysicsLM4
  Physics of Language Models, Part 4
  ☆260Updated 3 months ago