Library for serving Transformers models on Amazon SageMaker
A unified framework for scalable computing
Jittor is a high-performance deep learning framework
Powering Amazon custom machine learning chips
A ranked list of awesome machine learning Python libraries
PyTorch3D is FAIR's library of reusable components for deep learning
Graph Neural Network Library for PyTorch
Spatiotemporal Signal Processing with Neural Machine Learning Models
Ongoing research training transformer models at scale
A python library for self-supervised learning on images
Photorealistic Synthetic Dataset for Holistic Indoor Scene
AutoML library for deep learning
Our first fully AI generated deep learning system
Pre-trained Deep Learning models and demos
PyTorch code and models for V-JEPA self-supervised learning from video
An implementation of a deep learning recommendation model (DLRM)
Probabilistic reasoning and statistical analysis in TensorFlow
Making large AI models cheaper, faster and more accessible
An industrial grade federated learning framework
Package of deep-learning based CTR models
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
PyTorch code and models for VJEPA2 self-supervised learning from video
Petastorm library enables single machine or distributed training
Probabilistic time series modeling in Python