A tiny library for stochastic dataset caching in PyTorch.

(ICML 2023) High Fidelity Image Counterfactuals with Probabilistic Causal Models

Data descriptor and sample notebooks for the Emory Breast Imaging Dataset (EMBED) hosted on the AWS Open Data Program

RETFound - A foundation model for retinal image

[MICCAI 2023] "The Role of Subgroup Separability in Group-Fair Medical Image Classification"

A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.

understanding model mistakes with human annotations

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

An anomaly detection library comprising state-of-the-art algorithms and features such as experiment management, hyper-parameter optimization, and edge inference.

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

Laplace approximations for Deep Learning.

To Trust Or Not To Trust A Classifier. A measure of uncertainty for any trained (possibly black-box) classifier which is more effective than the classifier's own implied confidence (e.g. softmax pr…

Official repository for 'Algorithmic encoding of protected characteristics in chest X-ray disease detection models'

Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"

Addressing Failure Prediction by Learning Model Confidence

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

HI-ML toolbox for deep learning for medical imaging and Azure integration

Medical Imaging Deep Learning library to train and deploy 3D segmentation models on Azure Machine Learning