Open deep learning compiler stack for cpu, gpu and specialized accelerators

Productive, portable, and performant GPU programming in Python.

A list of awesome compiler projects and papers for tensor computation and deep learning.

A minimal GPU design in Verilog to learn how GPUs work from the ground up

CUDA Templates for Linear Algebra Subroutines

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

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

Datasets, Transforms and Models specific to Computer Vision

Assignment 1: automatic differentiation

PingCAP training courses

A PyTorch implementation of EfficientNet

PyTorch Framework for Developing Memory Efficient Deep Invertible Networks

(Spring 2018) Assignment 2: Graph Executor with TVM