Transformer-SRU LiDAR navigation for Unitree G1: Isaac Lab training, MuJoCo validation, and ROS 2 deployment.
:one: OneWM: Can World and Action Modeling Benefit Each Other under Unified Causal Training?
An autonomous agent that profiles, optimizes, and validates AI training code.
Efficient Mask-Aware Distributed Attention for Long-Context Training
ScaleNav: Scalable Internet Video Pretraining for Language-Conditioned Local Navigation
Simple study repo for tiny GPT-like model training (not a complete, functional model, for education only)
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Making large AI models cheaper, faster and more accessible
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework
Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.
Machine Learning Engineering Open Book
Ongoing research training transformer models at scale
RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.
Gorilla: Training and Evaluating LLMs for Function Calls (Tool Calls)
The ultimate training toolkit for finetuning diffusion models
A straightforward method for training your LLM, from downloading data to generating text.
Low-code framework for building custom LLMs, neural networks, and other AI models
Agent Reinforcement Trainer: train multi-step agents for real-world tasks using GRPO. Give your agents on-the-job training. Reinforcement learning for Qwen3.6, GPT-OSS, Llama, and more!
The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.
PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection with training, validation, inference, and multi-format export.
Build, Manage and Deploy AI/ML Systems
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
slime is an LLM training framework for RL Scaling.
Infrastructure for continually self‑improving agents
Data processing for and with foundation models! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷
《深入理解 AI Infra:量化分析与系统设计》(李博杰 著)开源书稿:从硬件约束和模型架构出发,量化推导 LLM 推理与训练系统设计。含全书正文、PDF、配套计算工具与实验
Efficient Triton Kernels for LLM Training
TSP自托管、零运维的 A 股「自建策略 + 监控 + 回测」量化工作台 | LLM能力驱使策略定制+个股分析+复盘 | 自由接入第三方数据源与个性化扩展数据 | 个人开源
SD-Trainer. LoRA & Dreambooth training scripts & GUI use kohya-ss's trainer, for diffusion model.
A system for quickly generating training data with weak supervision
Democratizing Reinforcement Learning for LLMs
A PyTorch native platform for training generative AI models
A Next-Generation Training Engine Built for Ultra-Large MoE Models
H2O LLM Studio - a framework and no-code GUI for fine-tuning LLMs. Documentation: https://docs.h2o.ai/h2o-llmstudio/
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
A unified inference and post-training framework for accelerated video generation.
⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.
Post-training with Tinker
AllenAI's post-training codebase
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools
A high-performance distributed training framework for Reinforcement Learning
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
LLM-as-a-Verifier is a general-purpose framework that provides fine-grained feedback for any agent without requiring additional training. It achieves SOTA performance across coding, robotics, and medical agentic benchmarks.
OneTrainer is a one-stop solution for all your Diffusion training needs.
Miles is an enterprise-facing reinforcement learning framework for LLM and VLM post-training, forked from and co-evolving with slime.
[NeurIPS 2026] DataFlex: A Unified Benchmark and Evaluation Platform for Data-Centric Training of Large Language Models
PyTorch native quantization for training and inference
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
Minimalistic large language model 3D-parallelism training
An interface library for RL post training with environments.
SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime
A nano replica of Jev: parallel decisions, dynamic candidates, and an end-to-end training pipeline.
Improve your Baduk skills by training with KataGo!
A Lightweight LLM Post-Training Library
RL training environments for Microduck (mjlab)