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