PyTorch extensions for fast R&D prototyping and Kaggle farming

A Python toolbox for conformal prediction research on deep learning models, using PyTorch.

Open-source framework for uncertainty and deep learning models in PyTorch 🌱

A scikit-learn compatible neural network library that wraps PyTorch

PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.

High quality, fast, modular reference implementation of SSD in PyTorch

A PyTorch native platform for training generative AI models

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

CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch

PyZMQ: Python bindings for zeromq

Uncertainty quantification with PyTorch

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

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

Accelerate PyTorch models with ONNX Runtime

PyTorch native quantization and sparsity for training and inference

PyTorch Extension Library of Optimized Scatter Operations