🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

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.

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

Image-to-Image Translation in PyTorch

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

End-to-End Object Detection with Transformers

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

Agent Reinforcement Trainer: train multi-step agents for real-world tasks using GRPO. Give your agents on-the-job training. Reinforcement learning for Qwen2.5, Qwen3, Llama, and more!

Synthesizing and manipulating 2048x1024 images with conditional GANs

Progressive Growing of GANs for Improved Quality, Stability, and Variation

PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.

An Open-Source Package for Knowledge Embedding (KE)

TextGrad: Automatic ''Differentiation'' via Text -- using large language models to backpropagate textual gradients. Published in Nature.

Code for reproducing results in "Glow: Generative Flow with Invertible 1x1 Convolutions"

Pointcept: Perceive the world with sparse points, a codebase for point cloud perception research. Latest works: Concerto (NeurIPS'25), Sonata (CVPR'25 Highlight), PTv3 (CVPR'24 Oral)

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!

Faster-RCNN in Tensorflow

Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"

Unsupervised Data Augmentation (UDA)

Generic U-Net Tensorflow implementation for image segmentation

Pytorch implementation of various Knowledge Distillation (KD) methods.

A python library and collection of notebooks for making art with machine learning.

Retina blood vessel segmentation with a convolutional neural network

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.

A Tensorflow implementation of Spatial Transformer Networks.