A Python framework for high performance GPU simulation and graphics

Puzzles for learning Triton

Ongoing research training transformer models at scale

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

全国各省市停贷通知汇总

深度学习经典、新论文逐段精读

The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement …

DaCe - Data Centric Parallel Programming

Swift for TensorFlow

Curated list of open-source quantum software projects.

World generator using simulation of plates, rain shadow, erosion, etc.

A Python framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.

PlaidML is a framework for making deep learning work everywhere.

"Multi-Level Intermediate Representation" Compiler Infrastructure

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

A code generator for array-based code on CPUs and GPUs

Dust3D is a cross-platform 3D modeling software that makes it easy to create low poly 3D models for video games, 3D printing, and more.

Tutorials for the Kokkos C++ Performance Portability Programming Ecosystem

CUDA Templates for Linear Algebra Subroutines

Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and primitives.

LLVM AMDGPU Assembler Helper Tools

x86-64 assembler embedded in Python

Acceleration package for neural networks on multi-core CPUs

Library for specialized dense and sparse matrix operations, and deep learning primitives.

ROCm Device Libraries

Collective communications library with various primitives for multi-machine training.

Simplex mesh adaptivity for HPC

TensorFlow ROCm port