Alternatives and similar repositories for batch_invariant_ops

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

• NVIDIA-NeMo / RL
  Scalable toolkit for efficient model reinforcement
  ☆1,267Updated this week
• 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
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆1,034Updated this week
• MoonshotAI / checkpoint-engine
  Checkpoint-engine is a simple middleware to update model weights in LLM inference engines
  ☆891Updated last week
• meta-pytorch / torchforge
  PyTorch-native post-training at scale
  ☆605Updated this week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)
  ☆781Updated last week
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆866Updated this week
• NovaSky-AI / SkyRL
  SkyRL: A Modular Full-stack RL Library for LLMs
  ☆1,518Updated this week
• NVIDIA-NeMo / Skills
  A project to improve skills of large language models
  ☆804Updated this week
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆334Updated 3 months ago
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆742Updated this week
• apple / ml-cross-entropy
  ☆579Updated 4 months ago
• thinking-machines-lab / tinker
  Training API and CLI
  ☆323Updated last week
• allenai / OLMo-core
  PyTorch building blocks for the OLMo ecosystem
  ☆763Updated this week
• huggingface / kernels
  Load compute kernels from the Hub
  ☆389Updated this week
• 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
• McGill-NLP / nano-aha-moment
  Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"
  ☆589Updated 3 months ago
• sail-sg / oat
  🌾 OAT: A research-friendly framework for LLM online alignment, including reinforcement learning, preference learning, etc.
  ☆621Updated 3 weeks ago
• NVIDIA-NeMo / Megatron-Bridge
  Training library for Megatron-based models with bidirectional Hugging Face conversion capability
  ☆400Updated this week
• huggingface / picotron_tutorial
  ☆230Updated 2 months ago
• facebookresearch / LayerSkip
  Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024
  ☆355Updated last week
• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆357Updated this week
• fla-org / native-sparse-attention
  🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"
  ☆964Updated 10 months ago
• mingyin0312 / RLFromScratch
  ☆465Updated 5 months ago
• QwenLM / ParScale
  Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scaling
  ☆469Updated 8 months ago
• microsoft / dion
  Dion optimizer algorithm
  ☆424Updated 2 weeks ago
• Dao-AILab / sonic-moe
  Accelerating MoE with IO and Tile-aware Optimizations
  ☆563Updated 2 weeks ago
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆341Updated 2 months ago
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆394Updated 7 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