Tensors and Dynamic neural networks in Python with strong GPU acceleration

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Datasets, Transforms and Models specific to Computer Vision

Pure bash script to test and wait on the availability of a TCP host and port

Deep universal probabilistic programming with Python and PyTorch

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Natural Gradient Boosting for Probabilistic Prediction

High-quality implementations of standard and SOTA methods on a variety of tasks.

Corruption and Perturbation Robustness (ICLR 2019)

Code for "Uncertainty Estimation Using a Single Deep Deterministic Neural Network"

Code to accompany the paper 'Improving model calibration with accuracy versus uncertainty optimization'.

Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning