Graph Neural Network Library for PyTorch
NVIDIA Federated Learning Application Runtime Environment
Petastorm library enables single machine or distributed training
Python package built to ease deep learning on graph
A python library for self-supervised learning on images
Build cross-modal and multimodal applications on the cloud
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling
High quality, fast, modular reference implementation of SSD in PyTorch
Embed images and sentences into fixed-length vectors
Serve machine learning models within a Docker container
The unified and scalable ML library for large-scale training
Fast Python collaborative filtering for implicit feedback datasets
An Open Toolkit for Knowledge Graph Extraction and Construction
Interactive deep learning book with multi-framework code
PyTorch Lightning + Hydra. A very user-friendly template
TF2 Deep FloorPlan Recognition using a Multi-task Network
Learning to rank in TensorFlow
Contextually-keyed word vectors
Lightweight anchor-free object detection model
Chinese-language edition of Dive into Deep Learning
Implementation of BEVFormer, a camera-only framework
Automatic architecture search and hyperparameter optimization
Data augmentation for NLP
Deep learning PyTorch library for time series forecasting
A flexible and efficient library for deep learning