Alternatives and similar repositories for torchforge

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

• radixark / miles
  ☆686Updated this week
• NVIDIA-NeMo / RL
  Scalable toolkit for efficient model reinforcement
  ☆1,210Updated this week
• thinking-machines-lab / batch_invariant_ops
  ☆945Updated 2 months ago
• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆343Updated 2 weeks ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆329Updated 2 months ago
• microsoft / dion
  Dion optimizer algorithm
  ☆413Updated this week
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆885Updated this week
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆306Updated last month
• huggingface / kernels
  Load compute kernels from the Hub
  ☆357Updated 3 weeks ago
• NVIDIA-NeMo / Automodel
  Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
  ☆232Updated this week
• thinking-machines-lab / tinker
  Training API and CLI
  ☆305Updated 3 weeks ago
• huggingface / picotron_tutorial
  ☆225Updated last month
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆985Updated this week
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆465Updated 2 weeks ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆277Updated last month
• facebookresearch / PhysicsLM4
  Physics of Language Models, Part 4
  ☆291Updated this week
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆267Updated this week
• tilde-research / MoMoE-impl
  Memory optimized Mixture of Experts
  ☆72Updated 5 months ago
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆700Updated this week
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆207Updated this week
• axon-rl / gem
  A Gym for Agentic LLMs
  ☆412Updated 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…
  ☆366Updated last year
• open-tinker / OpenTinker
  OpenTinker is an RL-as-a-Service infrastructure for foundation models
  ☆499Updated last week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)
  ☆732Updated this week
• mingyin0312 / RLFromScratch
  ☆464Updated 4 months ago
• allenai / OLMo-core
  PyTorch building blocks for the OLMo ecosystem
  ☆656Updated this week
• NVIDIA-NeMo / Megatron-Bridge
  HuggingFace conversion and training library for Megatron-based models
  ☆324Updated this week
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆740Updated 3 weeks ago
• pyember / ember
  ☆233Updated this week
• deepseek-ai / LPLB
  An early research stage expert-parallel load balancer for MoE models based on linear programming.
  ☆484Updated last month