AI Agent Skills

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

Datasets, Transforms and Models specific to Computer Vision

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

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

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, ...)

Rust bindings for the C++ api of PyTorch.

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

Simulation of spiking neural networks (SNNs) using PyTorch.