Python package built to ease deep learning on graph
Powering Amazon custom machine learning chips
The data structure for multimodal data
Implementation of Make-A-Video, new SOTA text to video generator
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A simple forecasting package
AutoGluon: AutoML for Image, Text, and Tabular Data
Library for serving Transformers models on Amazon SageMaker
An industrial grade federated learning framework
PyTorch3D is FAIR's library of reusable components for deep learning
A GPU-accelerated library containing highly optimized building blocks
Deep Learning Visualization Toolkit
Photorealistic Synthetic Dataset for Holistic Indoor Scene
Enabling PyTorch on Google TPU
A game theoretic approach to explain the output of ml models
ImageBind One Embedding Space to Bind Them All
Audiocraft is a library for audio processing and generation
OneFlow is a deep learning framework designed to be user-friendly
Simple and distributed Machine Learning
PyTorch code and models for V-JEPA self-supervised learning from video
An implementation of a deep learning recommendation model (DLRM)
Self-supervised visual learning using momentum contrast in PyTorch
Deep learning optimization library: makes distributed training easy
PyTorch code and models for VJEPA2 self-supervised learning from video
Best practices on recommendation systems