Alternatives and similar repositories for batch_invariant_ops

Users that are interested in batch_invariant_ops 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
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆985Updated this week
• NovaSky-AI / SkyRL
  SkyRL: A Modular Full-stack RL Library for LLMs
  ☆1,437Updated this week
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆885Updated last week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)
  ☆748Updated this week
• thinking-machines-lab / tinker
  Training API and CLI
  ☆305Updated 3 weeks ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆328Updated 2 months ago
• meta-pytorch / torchforge
  PyTorch-native post-training at scale
  ☆585Updated this week
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆707Updated this week
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆749Updated 3 weeks ago
• NVIDIA-NeMo / Skills
  A project to improve skills of large language models
  ☆756Updated this week
• McGill-NLP / nano-aha-moment
  Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"
  ☆576Updated 3 months ago
• apple / ml-cross-entropy
  ☆575Updated 3 months ago
• allenai / OLMo-core
  PyTorch building blocks for the OLMo ecosystem
  ☆681Updated this week
• huggingface / picotron_tutorial
  ☆224Updated last month
• sail-sg / oat
  🌾 OAT: A research-friendly framework for LLM online alignment, including reinforcement learning, preference learning, etc.
  ☆609Updated 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…
  ☆370Updated last year
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆278Updated last month
• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆343Updated 3 weeks ago
• MoonshotAI / Moonlight
  Muon is Scalable for LLM Training
  ☆1,397Updated 5 months ago
• QwenLM / ParScale
  Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scaling
  ☆467Updated 7 months ago
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆393Updated 6 months ago
• microsoft / dion
  Dion optimizer algorithm
  ☆413Updated last week
• huggingface / search-and-learn
  Recipes to scale inference-time compute of open models
  ☆1,123Updated 7 months ago
• HazyResearch / Megakernels
  kernels, of the mega variety
  ☆640Updated 3 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆467Updated 2 weeks ago
• zhuzilin / ring-flash-attention
  Ring attention implementation with flash attention
  ☆961Updated 4 months ago
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models with bi-directional Hugging Face conversion capability
  ☆347Updated this week
• mingyin0312 / RLFromScratch
  ☆465Updated 4 months ago