Stars
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
An open-source tool-augmented conversational language model from Fudan University
A Python implementation of global optimization with gaussian processes.
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams
Uplift modeling and causal inference with machine learning algorithms
Universal and Transferable Attacks on Aligned Language Models
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
Package for causal inference in graphs and in the pairwise settings. Tools for graph structure recovery and dependencies are included.
A plug-and-play library for parameter-efficient-tuning (Delta Tuning)
AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty
Prefix-Tuning: Optimizing Continuous Prompts for Generation
RobustBench: a standardized adversarial robustness benchmark [NeurIPS 2021 Benchmarks and Datasets Track]
Code relative to "Reliable evaluation of adversarial robustness with an ensemble of diverse parameter-free attacks"
DAGs with NO TEARS: Continuous Optimization for Structure Learning
The open-sourced Python toolbox for backdoor attacks and defenses.
Second-order Attention Network for Single Image Super-resolution (CVPR-2019)
PyTorch code to run synthetic experiments.
A every-so-often-updated collection of every causality + machine learning paper submitted to arXiv in the recent past.
Provable adversarial robustness at ImageNet scale
Backdoors Framework for Deep Learning and Federated Learning. A light-weight tool to conduct your research on backdoors.
Causal Effect Inference with Deep Latent-Variable Models
This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence Functions by Pang Wei Koh and Percy Liang.
TrojanZoo provides a universal pytorch platform to conduct security researches (especially backdoor attacks/defenses) of image classification in deep learning.