Train machine learning models within Docker containers
Trainable models and NN optimization tools
Implementation of Phenaki Video, which uses Mask GIT
A modular high-level library to train embodied AI agents
Probabilistic time series modeling in Python
Decentralized deep learning in PyTorch. Built to train models
Training and deploying machine learning models on Amazon SageMaker
Determined, deep learning training platform
Images to inference with no labeling
Train a 26M-parameter GPT from scratch in just 2h
Integrate, train and manage any AI models and APIs with your database
1 min voice data can also be used to train a good TTS model
The simplest, fastest repository for training/finetuning models
The most intuitive, flexible, way for researchers to build models
Implementation of Imagen, Google's Text-to-Image Neural Network
Awesome multilingual OCR toolkits based on PaddlePaddle
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Deep learning optimization library making distributed training easy
The machine learning toolkit for time series analysis in Python
Train multi-step agents for real-world tasks using GRPO
Toolkit for running TensorFlow training scripts on SageMaker
TextWorld is a sandbox learning environment for the training
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)