AI Agent Skills

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Time series forecasting with PyTorch

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Training PyTorch models with differential privacy

A lightweight library for PyTorch training tools and utilities

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Datasets, Transforms and Models specific to Computer Vision

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

On-device AI across mobile, embedded and edge for PyTorch

Deep universal probabilistic programming with Python and PyTorch

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

A scikit-learn compatible neural network library that wraps PyTorch

tensorboard for pytorch (and chainer, mxnet, numpy, ...)