DepGraph: Towards Any Structural Pruning
slime is an LLM post-training framework for RL Scaling
Sandbox for training deep learning networks
Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
RF-DETR is a real-time object detection and segmentation
Implementation of Vision Transformer, a simple way to achieve SOTA
An open-source, modern-design AI training tracking and visualization
High-Performance Face Recognition Library on PaddlePaddle & PyTorch
Ultralytics YOLO
[CVPR 2025 Best Paper Award] VGGT
A fast TTS architecture with conditional flow matching
PyTorch implementation of JiT
A simple, performant and scalable Jax LLM
dLLM: Simple Diffusion Language Modeling
Train any agents simply by 'talking'
TextWorld is a sandbox learning environment for the training
SDG is a specialized framework
RL implementations
A general fine-tuning kit geared toward image/video/audio diffusion
Ongoing research training transformer models at scale
The repository provides code for running inference with SAM 2
Unify Efficient Fine-tuning of RAG Retrieval, including Embedding
C++-based high-performance parallel environment execution engine
Open-source deep-learning framework for building and training
ICLR2024 Spotlight: curation/training code, metadata, distribution