Stars
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
📊 A simple command-line utility for querying and monitoring GPU status
PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
On the Variance of the Adaptive Learning Rate and Beyond
A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.
An open-source Python framework for hybrid quantum-classical machine learning.
[arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis
A New Padding Scheme: Partial Convolution based Padding
A simple method to perform semi-supervised learning with limited data.
My best practice of training large dataset using PyTorch.
Benchmarking Generalized Out-of-Distribution Detection
Code to create Stylized-ImageNet, a stylized version of standard ImageNet (ICLR 2019 Oral)
Open source release of the evaluation benchmark suite described in "Realistic Evaluation of Deep Semi-Supervised Learning Algorithms"
Code for the paper "A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attacks".
Repository for the CVPR19 oral paper "Domain Generalization by Solving Jigsaw Puzzles"
Official Pytorch Implementation for ICML'19 paper: Unsupervised Deep Learning by Neighbourhood Discovery
Reimplementation of "Realistic Evaluation of Deep Semi-Supervised Learning Algorithms"
Official implementation of Mitigating Sexual Content Generation via Embedding Distortion in Text-conditioned Diffusion Models
[ICCV 2025 Oral] Automated Model Evaluation for Object Detection via Prediction Consistency and Reliability
A standalone library to randomize various OpenAI Gym Environments
Network Randomization: A Simple Technique for Generalization in Deep Reinforcement Learning / ICLR 2020
[NeurIPS 2025] Official implementation of "Soft Task-Aware Routing of Experts for Equivariant Representation Learning"