AI Agent Skills

Accessible large language models via k-bit quantization for PyTorch.

A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.

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

Rust bindings for the C++ api of PyTorch.

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

Simulation of spiking neural networks (SNNs) using PyTorch.

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

A scala implementation of the Lightning Network.

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch

The PyTorch improved version of TPAMI 2017 paper: Face Alignment in Full Pose Range: A 3D Total Solution.

Siamese and triplet networks with online pair/triplet mining in PyTorch

Deep recommender models using PyTorch.