🤗 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.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Models and examples built with TensorFlow

《动手学深度学习》：面向中文读者、能运行、可讨论。中英文版被70多个国家的500多所大学用于教学。

Deep Learning for humans

The best ChatGPT that $100 can buy.

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

A book-in-progress about the Linux kernel and its insides.

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Graph Neural Network Library for PyTorch

Open standard for machine learning interoperability

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

Topic Modelling for Humans

Python Implementation of Reinforcement Learning: An Introduction

Python package built to ease deep learning on graph, on top of existing DL frameworks.

A MNIST-like fashion product database. Benchmark 👇

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

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

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

An open source python library for automated feature engineering

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

Convolutional Neural Network for Text Classification in Tensorflow

Model interpretability and understanding for PyTorch

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.

Web interface for browsing, search and filtering recent arxiv submissions

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.