Tensors and Dynamic neural networks in Python with strong GPU acceleration

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

🦛 CHONK docs with Chonkie ✨ — The no-nonsense RAG library

Like Vercel, but open source and for all languages.

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.