litian96

adversarial-robustness-toolbox

Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams

determined

Determined is an open-source machine learning platform that simplifies distributed training, hyperparameter tuning, experiment tracking, and resource management. Works with PyTorch and TensorFlow.

ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

loss-landscape

Code for visualizing the loss landscape of neural nets

mt-dnn

Multi-Task Deep Neural Networks for Natural Language Understanding

opacus

Training PyTorch models with differential privacy

pytorch_resnet_cifar10

Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.

FedProx

Federated Optimization in Heterogeneous Networks (MLSys '20)

arxiv-collector

A little Python script to collect LaTeX sources for upload to the arXiv.

mentornet

Code for MentorNet: Learning Data-Driven Curriculum for Very Deep Neural Networks

learning-to-reweight-examples

Code for paper "Learning to Reweight Examples for Robust Deep Learning"

backdoor_federated_learning

Source code for paper "How to Backdoor Federated Learning" (https://arxiv.org/abs/1807.00459)

fedjax

FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research.

fair_flearn

Fair Resource Allocation in Federated Learning (ICLR '20)

ffi-navigator

FedTree

A tree-based federated learning system (MLSys 2023)

fmtl

Federated Multi-Task Learning

CIDA

[ICML 2020] Continuously Indexed Domain Adaptation

TERM

Tilted Empirical Risk Minimization (ICLR '21)

fast-dpsgd

Code for fast dpsgd implementations in JAX/TF

Ditto

FedDANE

FedDANE: A Federated Newton-Type Method (Asilomar Conference on Signals, Systems, and Computers ‘19)

divfl

Diverse Client Selection for Federated Learning via Submodular Maximization

DP2

[ICLR 2023] Official JAX/Haiku implementation of the paper "Differentially Private Adaptive Optimization with Delayed Preconditioners"

AdaDPS

Private Adaptive Optimization with Side Information (ICML '22)

FedEx

pets-challenge

Binary_SVM_MTL

PCR_Release

Repository to go along with the paper "Progressive Compressed Records: Taking a Byte out of Deep Learning Data"

multi-criteria-dimensionality-reduction

Implementation and supporting materials for the paper Multi-Criteria Dimensionality Reduction with Applications to Fairness, NeurIPS2019 and extend the prior work "The Price of Fair PCA: One Extra Dimension"