The repository serves as a comprehensive model zoo encompassing a wide range of architectures including ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer variants, MobileNetV4, MobileNet-V3 and V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, and ConvNeXt. Beyond model definitions, the repository provides complete training, evaluation, inference, and export scripts alongside pretrained weights, making it a practical resource for practitioners implementing computer vision tasks.
The codebase is classified across multiple computer vision domains including semantic segmentation, object detection, image classification, and feature extraction, reflecting its broad applicability to diverse vision tasks.
Recent development activity shows intensive focus on Vision Transformer variants and emerging architectures. As of May 2026, the repository added EUPE ViT models with DINOv3-style training and ConvNeXt variants, along with TIPSv2 model definitions for DINOv2-style Vision Transformers. Earlier updates in 2025 introduced DINOv3 support for both ConvNeXt and ViT models, MobileCLIP-2 vision encoders, MetaCLIP-2 Worldwide ViT weights, and SigLIP-2 NaFlex ViT encoders. The repository also integrated support for Naver ROPE-ViT models and added MobileNetV5 backbone variants designed for Google Gemma 3n image encoding.
The codebase maintains active optimization efforts across multiple fronts. Recent releases introduced the Muon optimizer with customizations for convolutional weights and fallback mechanisms, alongside improvements to AdaMuon and NAdaMuon variants. Security enhancements include improved pickle checkpoint handling with weights_only=True as default and safe_global support for argument parsing. The repository added device and dtype factory keyword argument support across all models and modules, enabling flexible initialization strategies including meta-device model creation.
Benchmark coverage has expanded significantly, with new inference timing results added for RTX Pro 6000, 5090, and 4090 graphics cards using PyTorch 2.9.1. The repository maintains compatibility across PyTorch versions from 1.13 through 2.9.1 and Python versions from 3.10 through 3.13. Recent architectural additions include differential attention mechanisms, pooling modules like LsePlus and SimPool, and various normalization variants including Fp32 LayerNorm and RMSNorm options.
