PyTorch code and models for V-JEPA self-supervised learning from video
An implementation of a deep learning recommendation model (DLRM)
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
Medical imaging toolkit for deep learning
Easy-to-use,Modular and Extendible package of deep-learning models
Probabilistic reasoning and statistical analysis in TensorFlow
Graph Neural Network Library for PyTorch
An industrial grade federated learning framework
Package of deep-learning based CTR models
Open deep learning compiler stack for cpu, gpu, etc.
PyTorch code and models for VJEPA2 self-supervised learning from video
ktrain is a Python library that makes deep learning AI more accessible
A Python library for audio data augmentation
Petastorm library enables single machine or distributed training
Spatiotemporal Signal Processing with Neural Machine Learning Models
A refreshing functional take on deep learning
Probabilistic time series modeling in Python
AI Toolkit for Healthcare Imaging
AutoGluon: AutoML for Image, Text, and Tabular Data
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