Official code for ICLR 2023 paper "Towards a Unified Theoretical Understanding of Non-contrastive Learning via Rank Differential Mechanism"

PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566

This is the official impelementation of "FVSSL Algorithm"

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

pytorch memory track code

Benchmark of federated learning. Dedicated to the community. 🤗

Federated Learning Benchmark - Federated Learning on Non-IID Data Silos: An Experimental Study (ICDE 2022)

A small package to create visualizations of PyTorch execution graphs

torchview: visualize pytorch models

Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

[NeurIPS'23] FedL2P: Federated Learning to Personalize

Bayesian Nonparametric Federated Learning of Neural Networks

Code for Federated Learning with Matched Averaging, ICLR 2020.

Federated learning algorithms

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

FedorAS: Federated Architecture Search under system heterogeneity

ICLR 2022, "FedBABU: Toward Enhanced Representation for Federated Image Classification"

Comprehensive and timely academic information on federated learning (papers, frameworks, datasets, tutorials, workshops)

NVIDIA Federated Learning Application Runtime Environment

training food-101 (achieved SOTA top-1 validation acc ~=90%) using 1-cycle-policy:

37 traditional FL (tFL) or personalized FL (pFL) algorithms, 3 scenarios, and 24 datasets. www.pfllib.com/

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

[NeurIPS 2021]: Are Transformers More Robust Than CNNs? (Pytorch implementation & checkpoints)

My assignment solutions for CS231n - Convolutional Neural Networks for Visual Recognition

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561

Model-Contrastive Federated Learning (CVPR 2021)

Handy PyTorch implementation of Federated Learning (for your painless research)