Petastorm library enables single machine or distributed training
Implementation of Make-A-Video, new SOTA text to video generator
PyTorch code and models for V-JEPA self-supervised learning from video
A unified framework for scalable computing
Audiocraft is a library for audio processing and generation
On-device Speech-to-Intent engine powered by deep learning
Deep learning library
A python library for self-supervised learning on images
Pre-trained Deep Learning models and demos
ktrain is a Python library that makes deep learning AI more accessible
Jittor is a high-performance deep learning framework
Graph Neural Network Library for PyTorch
Powering Amazon custom machine learning chips
Determined, deep learning training platform
A refreshing functional take on deep learning
PyTorch code and models for VJEPA2 self-supervised learning from video
AutoGluon: AutoML for Image, Text, and Tabular Data
Library for serving Transformers models on Amazon SageMaker
Best practices on recommendation systems
Hub of ready-to-use datasets for ML models
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Open Source Computer Vision Library
Interactive deep learning book with multi-framework code
PyTorch Lightning + Hydra. A very user-friendly template
2D and 3D Face alignment library build using pytorch