PyTorch implementations of Generative Adversarial Networks.

Datasets, Transforms and Models specific to Computer Vision

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

Never use print for debugging again

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

End-to-End Object Detection with Transformers

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

Python Implementation of Reinforcement Learning: An Introduction

StyleGAN - Official TensorFlow Implementation

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

SciPy library main repository

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

100 numpy exercises (with solutions)

This repository contains the codes of "A Lip Sync Expert Is All You Need for Speech to Lip Generation In the Wild", published at ACM Multimedia 2020. For HD commercial model, please try out Sync Labs

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

An open-source NLP research library, built on PyTorch.

[CVPR 2025 Best Paper Award] VGGT: Visual Geometry Grounded Transformer

An educational resource to help anyone learn deep reinforcement learning.

Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

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

🐍 Geometric Computer Vision Library for Spatial AI

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

NumPy & SciPy for GPU

A little word cloud generator in Python

Refine high-quality datasets and visual AI models

TensorFlow-based neural network library

Record terminal sessions as SVG animations

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Simple Reinforcement learning tutorials, 莫烦Python 中文AI教学

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support