[ICCV 2025] Code for "Flow Stochastic Segmentation Networks"

Code for the paper "Robust representations for image classification via counterfactual contrastive learning" (Medical Image Analysis) and "Counterfactual contrastive learning: robust representation…

Code for the paper: Rotating Features for Object Discovery

A tiny library for stochastic dataset caching in PyTorch.

This is the official code release for Bayesian Flow Networks.

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

Convert Machine Learning Code Between Frameworks

Deep Bayesian Self-Training [official implementation]

counterfactuals for magnetic resonance images of multiple sclerosis

Implementations of Normalizing Flows in Pytorch/Pyro

Normalizing flows in PyTorch

Real NVP PyTorch a Minimal Working Example | Normalizing Flow

🎓 Path to a free self-taught education in Computer Science!

Official mirror of Python-FHEz; Python Fully Homomorphic Encryption (FHE) Library for Encrypted Deep Learning as a Service (EDLaaS).

Code for "Depth Uncertainty in Neural Networks" (https://arxiv.org/abs/2006.08437)

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

[ICLR'20] [PyTorch] Inverted Attention Routing for Capsules

Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

End-to-End Object Detection with Transformers

Image-to-Image Translation in PyTorch

A curated list of awesome work on VAEs, disentanglement, representation learning, and generative models.

Machine Learning Interviews from FAANG, Snapchat, LinkedIn. I have offers from Snapchat, Coupang, Stitchfix etc. Blog: mlengineer.io.

Literature survey, paper reviews, experimental setups and a collection of implementations for baselines methods for predictive uncertainty estimation in deep learning models.

Verification of ReZero ResNet on cifar 100 dataset

Pytorch code for managing distributed training experiments.