torch_musa is an open source repository based on PyTorch, which can make full use of the super computing power of MooreThreads graphics cards.

[CoRL'23] Parting with Misconceptions about Learning-based Vehicle Motion Planning

A JAX-based simulator for autonomous driving research.

SoftVC VITS Singing Voice Conversion

[CVPR 2023 Best Paper] Planning-oriented Autonomous Driving

Making large AI models cheaper, faster and more accessible

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

The devkit of the nuPlan dataset.

A latent text-to-image diffusion model

High-Resolution Image Synthesis with Latent Diffusion Models

[TNNLS] Differentiable Integrated Prediction and Planning Framework for Urban Autonomous Driving https://mczhi.github.io/DIPP/

MTR adaption: Motion Transformer with Global Intention Localization and Local Movement Refinement

[CVPR 2022] HiVT: Hierarchical Vector Transformer for Multi-Agent Motion Prediction

[ICLR 2022] DAB-DETR: Dynamic Anchor Boxes are Better Queries for DETR

CV backbones including GhostNet, TinyNet and TNT, developed by Huawei Noah's Ark Lab.

Tools for manipulating 3D meshes within the Menpo project.

Datasets, Transforms and Models specific to Computer Vision

Lip and hair color editor using face parsing maps.

[ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845

Final Project Repository for CMU's Learning Based Image Synthesis Course. Based on StyleGAN2-ADA - Official PyTorch implementation

Open Source Computer Vision Library

An Open Source Machine Learning Framework for Everyone

Reference implementations of popular deep learning models.

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

scikit-learn: machine learning in Python

Github Pages template for academic personal websites, forked from mmistakes/minimal-mistakes

A tree explorer plugin for vim.

Tensors and Dynamic neural networks in Python with strong GPU acceleration