An opinionated list of awesome Python frameworks, libraries, software and resources.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Models and examples built with TensorFlow

Official Code for DragGAN (SIGGRAPH 2023)

OpenMMLab Detection Toolbox and Benchmark

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Data science Python notebooks: Deep learning (TensorFlow, Theano, Caffe, Keras), scikit-learn, Kaggle, big data (Spark, Hadoop MapReduce, HDFS), matplotlib, pandas, NumPy, SciPy, Python essentials,…

Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

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

The interactive graphing library for Python ✨

Never use print for debugging again

Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).

End-to-End Object Detection with Transformers

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

Python bindings for FFmpeg - with complex filtering support

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

Refine high-quality datasets and visual AI models

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

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

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

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

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

A faster pytorch implementation of faster r-cnn

A library to generate LaTeX expression from Python code.

Object detection, 3D detection, and pose estimation using center point detection: