Alternatives and similar repositories for batch_invariant_ops

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

• thinking-machines-lab / tinker-cookbook
  Post-training with Tinker
  ☆550Updated this week
• NVIDIA-NeMo / RL
  Scalable toolkit for efficient model reinforcement
  ☆910Updated this week
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆751Updated this week
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆280Updated last month
• McGill-NLP / nano-aha-moment
  Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"
  ☆535Updated 2 months ago
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆650Updated this week
• huggingface / picotron_tutorial
  ☆221Updated 7 months ago
• NovaSky-AI / SkyRL
  SkyRL: A Modular Full-stack RL Library for LLMs
  ☆906Updated this week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆581Updated 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…
  ☆343Updated 9 months ago
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆623Updated this week
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆462Updated this week
• wolfecameron / nanoMoE
  An extension of the nanoGPT repository for training small MOE models.
  ☆195Updated 6 months ago
• NVIDIA / NeMo-Skills
  A project to improve skills of large language models
  ☆568Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆290Updated last week
• QwenLM / ParScale
  Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scaling
  ☆443Updated 4 months ago
• apple / ml-cross-entropy
  ☆531Updated last week
• sail-sg / oat
  🌾 OAT: A research-friendly framework for LLM online alignment, including reinforcement learning, preference learning, etc.
  ☆472Updated 2 weeks ago
• fla-org / native-sparse-attention
  🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"
  ☆877Updated 6 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…
  ☆268Updated 2 months ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆540Updated 4 months ago
• meta-pytorch / monarch
  PyTorch Single Controller
  ☆425Updated this week
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆219Updated this week
• mingyin0312 / RLFromScratch
  ☆435Updated last month
• facebookresearch / PhysicsLM4
  Physics of Language Models, Part 4
  ☆247Updated 2 months ago
• MoonshotAI / Moonlight
  Muon is Scalable for LLM Training
  ☆1,318Updated 2 months ago
• facebookresearch / LayerSkip
  Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024
  ☆338Updated 5 months ago
• open-thought / reasoning-gym
  [NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Verifiable Rewards
  ☆1,161Updated this week
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆296Updated last month
• microsoft / dion
  Dion optimizer algorithm
  ☆360Updated this week