A Lightweight LLM Post-Training Library
☆2,176Mar 9, 2026Updated this week
Alternatives and similar repositories for tunix
Users that are interested in tunix are comparing it to the libraries listed below
Sorting:
- a Jax quantization library☆94Updated this week
- A JAX-native High Performance Eval Metrics Library☆58Feb 3, 2026Updated last month
- Library for reading and processing ML training data.☆688Updated this week
- Minimal, lightweight JAX implementations of popular models.☆207Updated this week
- A PyTorch native platform for training generative AI models☆5,111Updated this week
- Our library for RL environments + evals☆3,877Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆247Updated this week
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated 9 months ago
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- PyTorch native post-training library☆5,697Updated this week
- Agent Reinforcement Trainer: train multi-step agents for real-world tasks using GRPO. Give your agents on-the-job training. Reinforcement…☆8,976Updated this week
- A simple, performant and scalable Jax LLM!☆2,163Updated this week
- verl: Volcano Engine Reinforcement Learning for LLMs☆19,739Updated this week
- Open-source context retrieval layer for AI agents☆5,901Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,870Jun 22, 2025Updated 8 months ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆414Jan 5, 2026Updated 2 months ago
- JAX backend for SGL☆243Mar 3, 2026Updated last week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆550Feb 26, 2026Updated last week
- Flax is a neural network library for JAX that is designed for flexibility.☆7,105Updated this week
- Post-training with Tinker☆2,901Updated this week
- A Datacenter Scale Distributed Inference Serving Framework☆6,193Updated this week
- Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.☆4,755Jul 18, 2025Updated 7 months ago
- Official inference framework for 1-bit LLMs☆28,697Feb 3, 2026Updated last month
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆5,330Updated this week
- Efficient Triton Kernels for LLM Training☆6,189Updated this week
- A community driven registry service for Model Context Protocol (MCP) servers.☆6,510Mar 3, 2026Updated last week
- Fine-tuning & Reinforcement Learning for LLMs. 🦥 Train OpenAI gpt-oss, DeepSeek, Qwen, Llama, Gemma, TTS 2x faster with 70% less VRAM.☆53,029Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,987Updated this week
- Easily fine-tune, evaluate and deploy gpt-oss, Qwen3, DeepSeek-R1, or any open source LLM / VLM!☆8,903Updated this week
- Modular, scalable library to train ML models☆216Updated this week
- AllenAI's post-training codebase☆3,614Updated this week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆695Jan 26, 2026Updated last month
- SGLang is a high-performance serving framework for large language models and multimodal models.☆24,216Updated this week
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated last week
- The conversational control layer for customer-facing AI agents - Parlant is a context-engineering framework optimized for controlling cus…☆17,799Updated this week
- Tongyi Deep Research, the Leading Open-source Deep Research Agent☆18,337Feb 27, 2026Updated last week
- DSPy: The framework for programming—not prompting—language models☆32,519Updated this week
- Fast, flexible LLM inference☆6,653Feb 27, 2026Updated last week
- Optax is a gradient processing and optimization library for JAX.☆2,198Feb 26, 2026Updated last week