Trainable models and NN optimization tools
Train machine learning models within Docker containers
Toolkit for running TensorFlow training scripts on SageMaker
Large Language Model Principles and Practice Tutorial from Scratch
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
Deep learning optimization library making distributed training easy
A Next-Generation Training Engine Built for Ultra-Large MoE Models
PyTorch building blocks for the OLMo ecosystem
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
State-of-the-art 2D and 3D Face Analysis Project
Unified web UI for training and running open models locally
A general fine-tuning kit geared toward image/video/audio diffusion
A lightweight library for PyTorch training tools and utilities
slime is an LLM post-training framework for RL Scaling
Supercharge Your Model Training
The simplest, fastest repository for training/finetuning models
Continual learning infra for self-improving agents
Understanding of AI Infra: Quantitative Analysis and System Design
Powerful AI language model (MoE) optimized for efficiency/performance
Training PyTorch models with differential privacy
PaddlePaddle End-to-End Development Toolkit
A straightforward method for training your LLM
Ongoing research training transformer models at scale
Reference PyTorch implementation and models for DINOv3
tensorboard for pytorch (and chainer, mxnet, numpy, etc.)