Hadi Salman's starred repositories
pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
PythonRobotics
Python sample codes for robotics algorithms.
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
img2dataset
Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.
auto-attack
Code relative to "Reliable evaluation of adversarial robustness with an ensemble of diverse parameter-free attacks"
robustbench
RobustBench: a standardized adversarial robustness benchmark [NeurIPS'21 Benchmarks and Datasets Track]
safety-gym
Tools for accelerating safe exploration research.
fast_adversarial
[ICLR 2020] A repository for extremely fast adversarial training using FGSM
ciphercore
User-friendly secure computation engine based on secure multi-party computation
semisup-adv
Semisupervised learning for adversarial robustness https://arxiv.org/pdf/1905.13736.pdf
ot_estimators
Implementations of several fast approximate algorithms for geometric optimal transport (OT)
flowattack
Attacking Optical Flow (ICCV 2019)
Transformer-from-scratch
I will build Transformer from scratch
failure-directions
Distilling Model Failures as Directions in Latent Space
Denoised-Smoothing-TF
Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.
improved_wasserstein
Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"