NVlabs/RADIO

Reference PyTorch implementation and models for DINOv3

PyTorch code and models for the DINOv2 self-supervised learning method.

OpenMMLab Pre-training Toolbox and Benchmark

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Scenic: A Jax Library for Computer Vision Research and Beyond

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

Recipes for shrinking, optimizing, customizing cutting edge vision models. 💜