Alternatives and similar repositories for torchforge

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

• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆357Updated this week
• thinking-machines-lab / tinker
  Training API and CLI
  ☆323Updated this week
• NVIDIA-NeMo / RL
  Scalable toolkit for efficient model reinforcement
  ☆1,267Updated this week
• thinking-machines-lab / batch_invariant_ops
  ☆952Updated 2 months ago
• radixark / miles
  Miles is an enterprise-facing reinforcement learning framework for LLM and VLM post-training, forked from and co-evolving with slime.
  ☆789Updated this week
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆333Updated 2 months ago
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆307Updated last month
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆891Updated last week
• NVIDIA-NeMo / Automodel
  Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
  ☆259Updated this week
• microsoft / dion
  Dion optimizer algorithm
  ☆420Updated 2 weeks ago
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models with bidirectional Hugging Face conversion capability
  ☆386Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆381Updated this week
• huggingface / picotron_tutorial
  ☆229Updated 2 months ago
• tilde-research / MoMoE-impl
  Memory optimized Mixture of Experts
  ☆72Updated 6 months ago
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆1,020Updated last week
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆280Updated 2 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆472Updated 2 weeks ago
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆370Updated last year
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆272Updated this week
• axon-rl / gem
  A Gym for Agentic LLMs
  ☆437Updated last week
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆731Updated this week
• allenai / OLMo-core
  PyTorch building blocks for the OLMo ecosystem
  ☆741Updated this week
• NVIDIA-NeMo / Skills
  A project to improve skills of large language models
  ☆786Updated this week
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆858Updated last week
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆223Updated this week
• sgl-project / sglang-jax
  JAX backend for SGL
  ☆232Updated this week
• apple / ml-cross-entropy
  ☆579Updated 4 months ago
• apple / ml-recurrent-drafter
  ☆219Updated last year
• sail-sg / oat
  🌾 OAT: A research-friendly framework for LLM online alignment, including reinforcement learning, preference learning, etc.
  ☆621Updated 3 weeks ago
• z-lab / dflash
  Block Diffusion for Ultra-Fast Speculative Decoding
  ☆432Updated this week