zhengpingzhou
Follow
Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
Rich is a Python library for rich text and beautiful formatting in the terminal.
A toolkit for developing and comparing reinforcement learning algorithms.
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Image-to-Image Translation in PyTorch
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Graph Neural Network Library for PyTorch
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
Datasets, Transforms and Models specific to Computer Vision
OpenAI Baselines: high-quality implementations of reinforcement learning algorithms
End-to-End Object Detection with Transformers
pix2code: Generating Code from a Graphical User Interface Screenshot
Python bindings for FFmpeg - with complex filtering support
A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
Faster R-CNN (Python implementation) -- see https://github.com/ShaoqingRen/faster_rcnn for the official MATLAB version
Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
An elegant PyTorch deep reinforcement learning library.
Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
PyTorch implementations of deep reinforcement learning algorithms and environments
A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.
Count the MACs / FLOPs of your PyTorch model.
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Reading Wikipedia to Answer Open-Domain Questions