Trainable models and NN optimization tools
Train machine learning models within Docker containers
A straightforward method for training your LLM
Build a modern LLM from scratch. Every line commented
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Train a 26M-parameter GPT from scratch in just 2h
Build and train a GPT-style language model
Determined, deep learning training platform
Probabilistic time series modeling in Python
Making large AI models cheaper, faster and more accessible
Decentralized deep learning in PyTorch. Built to train models
Fast State-of-the-Art Tokenizers optimized for Research and Production
Designed for training LLM/VLM agents via RL
Deepfakes Software For All
Graph Neural Network Library for PyTorch
The standard data-centric AI package for data quality and ML
The largest collection of PyTorch image encoders / backbones
1 min voice data can also be used to train a good TTS model
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Gradient boosting framework based on decision tree algorithms
Train multi-step agents for real-world tasks using GRPO
Machine Learning Toolkit for Kubernetes
"Big Model" trains a visual multimodal VLM with 26M parameters
RF-DETR is a real-time object detection and segmentation