Geometric deep learning extension library for PyTorch
Python-based neural networks API
AI Toolkit for Healthcare Imaging
Han Language Processing
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Petastorm library enables single machine or distributed training
An implementation of a deep learning recommendation model (DLRM)
Deep universal probabilistic programming with Python and PyTorch
Implementation of Make-A-Video, new SOTA text to video generator
Photorealistic Synthetic Dataset for Holistic Indoor Scene
PaddlePaddle End-to-End Development Toolkit
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
A game theoretic approach to explain the output of ml models
Ongoing research training transformer models at scale
Deep learning optimization library: makes distributed training easy
A ranked list of awesome machine learning Python libraries
Medical imaging toolkit for deep learning
Industrial-strength Natural Language Processing (NLP)
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
Library for serving Transformers models on Amazon SageMaker
MII makes low-latency and high-throughput inference possible