Petastorm library enables single machine or distributed training
A refreshing functional take on deep learning
A unified framework for scalable computing
Pre-trained Deep Learning models and demos
PyTorch code and models for V-JEPA self-supervised learning from video
On-device Speech-to-Intent engine powered by deep learning
A python library for self-supervised learning on images
Deep learning library
Audiocraft is a library for audio processing and generation
ktrain is a Python library that makes deep learning AI more accessible
Powering Amazon custom machine learning chips
Jittor is a high-performance deep learning framework
Hub of ready-to-use datasets for ML models
Graph Neural Network Library for PyTorch
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Determined, deep learning training platform
PyTorch code and models for VJEPA2 self-supervised learning from video
Self-supervised visual learning using momentum contrast in PyTorch
AutoGluon: AutoML for Image, Text, and Tabular Data
Library for serving Transformers models on Amazon SageMaker
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
Distributed training framework for TensorFlow, Keras, PyTorch, etc.