Alternatives and similar repositories for torchforge

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

• radixark / miles
  Miles is an enterprise-facing reinforcement learning framework for LLM and VLM post-training, forked from and co-evolving with slime.
  ☆830Updated this week
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆902Updated last week
• thinking-machines-lab / tinker
  Training API and CLI
  ☆325Updated last week
• NVIDIA-NeMo / RL
  Scalable toolkit for efficient model reinforcement
  ☆1,293Updated this week
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆334Updated 3 months ago
• thinking-machines-lab / batch_invariant_ops
  ☆957Updated 3 months ago
• NVIDIA-NeMo / Automodel
  Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
  ☆266Updated this week
• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆361Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆397Updated this week
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆475Updated this week
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆307Updated 2 months ago
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆273Updated this week
• huggingface / picotron_tutorial
  ☆232Updated 2 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆279Updated 2 months ago
• NVlabs / vibetensor
  Our first fully AI generated deep learning system
  ☆481Updated last week
• microsoft / dion
  Dion optimizer algorithm
  ☆424Updated 3 weeks ago
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆228Updated this week
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆1,044Updated this week
• NVIDIA-NeMo / Skills
  A project to improve skills of large language models
  ☆813Updated this week
• apple / ml-recurrent-drafter
  ☆219Updated last year
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆371Updated last year
• sgl-project / sglang-jax
  JAX backend for SGL
  ☆234Updated this week
• Dao-AILab / sonic-moe
  Accelerating MoE with IO and Tile-aware Optimizations
  ☆569Updated 3 weeks ago
• deepseek-ai / LPLB
  An early research stage expert-parallel load balancer for MoE models based on linear programming.
  ☆495Updated 2 months ago
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)
  ☆792Updated 2 weeks ago
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆876Updated last week
• z-lab / dflash
  Block Diffusion for Ultra-Fast Speculative Decoding
  ☆459Updated this week
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models with bidirectional Hugging Face conversion capability
  ☆419Updated this week
• tilde-research / MoMoE-impl
  Memory optimized Mixture of Experts
  ☆73Updated 6 months ago
• apple / ml-cross-entropy
  ☆579Updated 4 months ago