Training PyTorch models with differential privacy
APACHE-2.0 License
torchao: PyTorch Architecture Optimization (AO). Performant kernels that work with PyTorch.
Model interpretability and understanding for PyTorch
Collection of SOTA efficient computer vision models for embedded applications, with pre-trained w...
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Se...
PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for...
PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models ...
OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
A highly efficient implementation of Gaussian Processes in PyTorch
Run PyTorch LLMs locally on servers, desktop and mobile
Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™
Research Framework for easy and efficient training of GANs based on Pytorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...
Learn to make machines learn so that you don't have to struggle to program them; The ultimate list