An implementation of a deep learning recommendation model (DLRM)
PyTorch3D is FAIR's library of reusable components for deep learning
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
Medical imaging toolkit for deep learning
Probabilistic reasoning and statistical analysis in TensorFlow
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
An industrial grade federated learning framework
Geometric deep learning extension library for PyTorch
PyTorch code and models for VJEPA2 self-supervised learning from video
Audiocraft is a library for audio processing and generation
ktrain is a Python library that makes deep learning AI more accessible
A Python library for audio data augmentation
Probabilistic time series modeling in Python
AutoGluon: AutoML for Image, Text, and Tabular Data
A game theoretic approach to explain the output of ml models
Library for serving Transformers models on Amazon SageMaker
Python package built to ease deep learning on graph
Best practices on recommendation systems
MII makes low-latency and high-throughput inference possible
The data structure for multimodal data
Jittor is a high-performance deep learning framework
Fast image augmentation library and an easy-to-use wrapper
Build cross-modal and multimodal applications on the cloud