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

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

Federated learning algorithms

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

FedorAS: Federated Architecture Search under system heterogeneity

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)

Flower: A Friendly Federated AI Framework

Perform data science on data that remains in someone else's server

FedML - The Research and Production Integrated Federated Learning Library: https://fedml.ai