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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Development repository for the Triton language and compiler

A model compilation solution for various hardware

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.

GPGPU-Sim provides a detailed simulation model of contemporary NVIDIA GPUs running CUDA and/or OpenCL workloads. It includes support for features such as TensorCores and CUDA Dynamic Parallelism as…

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Primary repository for the Trilinos Project

An MLIR-based compiler framework bridges DSLs (domain-specific languages) to DSAs (domain-specific architectures).

LAPACK development repository

Reusable software components for rocm developers

Cooperative primitives for CUDA C++.

BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.

CUDA Library Samples

CUDA Templates for Linear Algebra Subroutines

NVIDIA® TensorRT™, an SDK for high-performance deep learning inference, includes a deep learning inference optimizer and runtime that delivers low latency and high throughput for inference applicat…

SparseTIR: Sparse Tensor Compiler for Deep Learning

浙江大学课程攻略共享计划

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs

CUSP : A C++ Templated Sparse Matrix Library

A home for the final text of all TVM RFCs.

清华大学计算机系课程攻略 Guidance for courses in Department of Computer Science and Technology, Tsinghua University

Performance Prediction Toolkit

CMSIS Version 5 Development Repository

A lightweight open-source C compiler for research and education.

CUDAAdvisor: a GPU profiling tool

Buddy memory allocation