Trainable models and NN optimization tools
Train machine learning models within Docker containers
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Train a 26M-parameter GPT from scratch in just 2h
Language modeling in a sentence representation space
Determined, deep learning training platform
Images to inference with no labeling
Probabilistic time series modeling in Python
1 min voice data can also be used to train a good TTS model
Making large AI models cheaper, faster and more accessible
Decentralized deep learning in PyTorch. Built to train models
"Big Model" trains a visual multimodal VLM with 26M parameters
A coding-free framework built on PyTorch
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Image inpainting tool powered by SOTA AI Model
On-device Speech-to-Intent engine powered by deep learning
The standard data-centric AI package for data quality and ML
Deep learning optimization library making distributed training easy
Towards Real-World Vision-Language Understanding
The fast, Pythonic way to build Model Context Protocol servers
Code for running inference and finetuning with SAM 3 model
The largest collection of PyTorch image encoders / backbones
Deepfakes Software For All
Z80-μLM is a 2-bit quantized language model
Graph Neural Network Library for PyTorch