rwightman

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

pytorch_geometric

Graph Neural Network Library for PyTorch

vit-pytorch

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

deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications

vision_transformer

flax

Flax is a neural network library for JAX that is designed for flexibility.

DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

lambda-networks

Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute

optax

Optax is a gradient processing and optimization library for JAX.

the-pile

awesome-jax

JAX - A curated list of resources https://github.com/google/jax

torchdistill

A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

VMZ

VMZ: Model Zoo for Video Modeling

tf-gnn-samples

TensorFlow implementations of Graph Neural Networks

objax

MEAL-V2

MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.

TimeSformer-pytorch

Implementation of TimeSformer from Facebook AI, a pure attention-based solution for video classification

robustness_metrics

self_supervised

Implementation of popular SOTA self-supervised learning algorithms as Fastai Callbacks.

rigl

End-to-end training of sparse deep neural networks with little-to-no performance loss.

task_adaptation

timmdocs

Documentation for Ross Wightman's timm image model library

PAA

A PyTorch implementation of the paper `Probabilistic Anchor Assignment with IoU Prediction for Object Detection` ECCV 2020 (https://arxiv.org/abs/2007.08103)

halonet-pytorch

Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones

glom-pytorch

An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up processing, and attention (consensus between columns), for emergent part-whole heirarchies from data

defgrid-release

Official PyTorch implementation of Deformable Grid (ECCV 2020)

baby-vision

Self-supervised learning through the eyes of a child

imagenet-testbed

ImageNet Testbed, associated with the paper "Measuring Robustness to Natural Distribution Shifts in Image Classification."

global-self-attention-network

A Pytorch implementation of Global Self-Attention Network, a fully-attention backbone for vision tasks

SeesawLoss