torch-uncertainty

Uncertainty tools

Provides tools and methods to quantify uncertainty in deep learning models and improve their reliability

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

GitHub

320 stars
9 watching
20 forks
Language: Python
last commit: about 1 month ago
bayesian-networkcomputer-visionensemblesmixupneural-networkspredictive-uncertaintypytorchreliable-aitrustworthy-machine-learninguncertaintyuncertainty-quantification

Related projects:

Repository Description Stars
sheffieldml/deepgp Software implementing probabilistic models for uncertainty estimation in data analysis 91
awslabs/fortuna A library for estimating uncertainty in machine learning models and predictions 892
azure/pyrit Empowers security professionals to identify risks in generative AI systems by providing a framework for proactive risk assessment and red teaming. 1,977
aiqm/torchani A PyTorch implementation of a neural network potential for molecular simulations 471
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,044
embodiedgpt/embodiedgpt_pytorch A PyTorch-based toolkit for creating customized multimedia datasets and handling heterogeneous data for training AI models. 346
akanimax/pro_gan_pytorch Implementation of a deep learning model for generating high-quality images with improved stability and variation. 538
ctallec/pyvarinf A Python package facilitating Bayesian Deep Learning methods with Variational Inference for PyTorch 359
project-dc/pygeneses A PyTorch-based framework for training and studying artificial species in bio-inspired environments 72
probtorch/probtorch A deep learning framework for probabilistic models that extends PyTorch with support for reparameterized distributions and Monte Carlo estimators. 887
jacobgil/pytorch-pruning This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. 877
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
ahmedfgad/torchga Trains PyTorch models using a genetic algorithm 96
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
jpeg729/pytorch_bits An experimental framework for developing and testing deep learning models on time-series prediction tasks 79