Taichi compatibility#

2025-07-31

4 min read time

Applies to Linux

Taichi is an open-source, imperative, and parallel programming language designed for high-performance numerical computation. Embedded in Python, it leverages just-in-time (JIT) compilation frameworks such as LLVM to accelerate compute-intensive Python code by compiling it to native GPU or CPU instructions.

Taichi is widely used across various domains, including real-time physical simulation, numerical computing, augmented reality, artificial intelligence, computer vision, robotics, visual effects in film and gaming, and general-purpose computing.

  • ROCm support for Taichi is hosted in the official ROCm/taichi repository.

  • Due to independent compatibility considerations, this location differs from the taichi-dev upstream repository.

  • Use the prebuilt Docker image with ROCm, PyTorch, and Taichi preinstalled.

  • See the ROCm Taichi installation guide to install and get started.

Note

Taichi is supported on ROCm 6.3.2.

Supported devices and features#

There is support through the ROCm software stack for all Taichi GPU features on AMD Instinct MI250X and MI210X series GPUs with the exception of Taichi’s GPU rendering system, CGUI. AMD Instinct MI300X series GPUs will be supported by November.

Use cases and recommendations#

To fully leverage Taichi’s performance capabilities in compute-intensive tasks, it is best to adhere to specific coding patterns and utilize Taichi decorators. A collection of example use cases is available in the ROCm/taichi_examples repository, providing practical insights and foundational knowledge for working with the Taichi programming language. You can also refer to the AMD ROCm blog to search for Taichi examples and best practices to optimize your workflows on AMD GPUs.

Docker image compatibility#

AMD validates and publishes ready-made ROCm Taichi Docker images with ROCm backends on Docker Hub. The following Docker image tags and associated inventories represent the latest Taichi version from the official Docker Hub. The Docker images have been validated for ROCm 6.3.2. Click to view the image on Docker Hub.

Docker image

ROCm

Taichi

Ubuntu

Python

rocm/taichi

6.3.2

1.8.0b1

22.04

3.10.12