🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

Your data - interactive!

Code for the CVPR 2023 paper "Proximal Splitting Adversarial Attacks for Semantic Segmentation"

PyTorch Image Quality Assessement package

Library containing PyTorch implementations of various adversarial attacks and resources

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

(CVPR 2021) Code for our method RePRI for Few-Shot Segmentation. Paper at http:https://arxiv.org/abs/2012.06166

Code for the ICCV 2021 paper "Augmented Lagrangian Adversarial Attacks"

A Python-embedded modeling language for convex optimization problems.

A Toolbox for Adversarial Robustness Research

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Latex code for making neural networks diagrams

A pytorch port of google-research/google-research/robust_loss/

Code for the CVPR 2019 article "Decoupling Direction and Norm for Efficient Gradient-Based L2 Adversarial Attacks and Defenses"

Wrapper to use visdom as logger when training ML models

A zoo of PyPlot displays examples