AI Agent Skills

Data manipulation and transformation for audio signal processing, powered by PyTorch

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

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

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

A PyTorch-based Speech Toolkit

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

Deep universal probabilistic programming with Python and PyTorch

Accessible large language models via k-bit quantization for 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, ...)

Code Repository for Machine Learning with PyTorch and Scikit-Learn

Dockerized FastAPI wrapper for Kokoro-82M text-to-speech model w/multiplatform CPU, AMD, NVIDIA GPU PyTorch support, handling, and auto-stitching