Let us control diffusion models!

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

A booklet on machine learning systems design with exercises. NOT the repo for the book "Designing Machine Learning Systems"

An educational resource to help anyone learn deep reinforcement learning.

A customized version of Multipath++ for object behavior prediction in Carla simulator

Copy-paste augmentation for segmentation and detection tasks

Graph Neural Network Library for PyTorch

Classical equations and diagrams in machine learning

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

Data collection for module Object Behavior Prediction in carla simulation

Classical equations and diagrams in machine learning