Stars
Minimal PyTorch implementation of YOLOv3
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
Library for building WebSocket servers and clients in Python
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Image augmentation library in Python for machine learning.
Count the MACs / FLOPs of your PyTorch model.
Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.
Torchreid: Deep learning person re-identification in PyTorch.
SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask.
A Python toolbox for performing gradient-free optimization
Machine Learning Pipelines for Kubeflow
Efficient 3D human pose estimation in video using 2D keypoint trajectories
GLIDE: a diffusion-based text-conditional image synthesis model
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"
The author's officially unofficial PyTorch BigGAN implementation.
An optimizer that trains as fast as Adam and as good as SGD.
Plenoxels: Radiance Fields without Neural Networks
A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
Pytorch framework for doing deep learning on point clouds.
Simultaneous object detection and tracking using center points.
pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593