Graph Neural Network Library for PyTorch
An industrial grade federated learning framework
Open deep learning compiler stack for cpu, gpu, etc.
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
A library for deep learning end-to-end dialog systems and chatbots
Deep learning library
On-device Speech-to-Intent engine powered by deep learning
Powering Amazon custom machine learning chips
OpenMMLab Model Deployment Framework
Open Source Computer Vision Library
A computer vision framework to create and deploy apps in minutes
2D and 3D Face alignment library build using pytorch
PyTorch Lightning + Hydra. A very user-friendly template
Interactive deep learning book with multi-framework code
A large open dataset + tools to speed up MRI scans using ML
Distributed training framework for TensorFlow, Keras, PyTorch, etc.
Graph Neural Networks with Keras and Tensorflow 2
Learning to rank in TensorFlow
The AI Agent Host is a module-based development environment.
A library for graph deep learning research
Implementation of model parallel autoregressive transformers on GPUs
Package of deep-learning based CTR models