A Lightweight LLM Post-Training Library
☆2,202Mar 29, 2026Updated this week
Alternatives and similar repositories for tunix
Users that are interested in tunix are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- a Jax quantization library☆102Mar 20, 2026Updated last week
- Multi-Turn RL Training System with AgentTrainer for Language Model Game Reinforcement Learning☆60Dec 18, 2025Updated 3 months ago
- A JAX-native High Performance Eval Metrics Library☆58Feb 3, 2026Updated last month
- Minimal, lightweight JAX implementations of popular models.☆214Mar 4, 2026Updated 3 weeks ago
- Minimal yet performant LLM examples in pure JAX☆246Jan 14, 2026Updated 2 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- Library for reading and processing ML training data.☆703Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆266Updated this week
- A simple, performant and scalable Jax LLM!☆2,182Mar 21, 2026Updated last week
- JAX backend for SGL☆252Mar 23, 2026Updated last week
- Post-training with Tinker☆2,971Mar 23, 2026Updated last week
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated 9 months ago
- A PyTorch native platform for training generative AI models☆5,191Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆550Mar 17, 2026Updated last week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆417Jan 5, 2026Updated 2 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Open-source context retrieval layer for AI agents☆6,058Updated this week
- Our library for RL environments + evals☆3,930Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,872Jun 22, 2025Updated 9 months ago
- PyTorch native post-training library☆5,713Updated this week
- verl: Volcano Engine Reinforcement Learning for LLMs☆20,286Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.☆7,136Updated this week
- ☆319Updated this week
- Modular, scalable library to train ML models☆227Updated this week
- Agent Reinforcement Trainer: train multi-step agents for real-world tasks using GRPO. Give your agents on-the-job training. Reinforcement…☆9,077Updated this week
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- slime is an LLM post-training framework for RL Scaling.☆4,906Mar 23, 2026Updated last week
- Optax is a gradient processing and optimization library for JAX.☆2,221Mar 20, 2026Updated last week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,190Mar 23, 2026Updated last week
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated 3 weeks ago
- Efficient Triton Kernels for LLM Training☆6,242Updated this week
- ☆278Updated this week
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆5,432Updated this week
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆172Updated this week
- Unsloth Studio is a web UI for training and running open models like Qwen, DeepSeek, gpt-oss and Gemma locally.☆57,673Mar 23, 2026Updated last week
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- A Datacenter Scale Distributed Inference Serving Framework☆6,411Updated this week
- Official inference framework for 1-bit LLMs☆36,743Mar 10, 2026Updated 2 weeks ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆701Jan 26, 2026Updated 2 months ago
- AllenAI's post-training codebase☆3,643Mar 23, 2026Updated last week
- [NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Verifiable Rewards☆1,375Updated this week
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆881Mar 15, 2026Updated 2 weeks ago
- Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.☆4,752Jul 18, 2025Updated 8 months ago