gstaichi 4.6.0

The GsTaichi Programming Language

GsTaichi

GsTaichi was forked in June 2025. This repository (or gstaichi) is now a fully independent project with no intention of maintaining backward compatibility with the original taichi. Whilst the repo largely resembles upstream for now, we have made the following changes:

• revamped continuous integration, to run using recent python versions (up to 3.13), recent mac os x versions (up to 15), and to run reliably (at least 80% of runs with correct code succeed)
• added dataclasses.dataclass structs:
  • work with both ndarrays and fields (cf ti.struct (field only), ti.dataclass (field only), ti.data_oriented (field only), argpack (ndarray only))
  • can be passed into child ti.functions (cf argpack)
  • does not affect kernel runtime speed (kernels see only the underlying arrays, no indirection is added within the kernel layer)
• removed GUI/GGUI, C-API, AOT, DX11, DX12, IOS, Android, OpenGL, GLES

Planned features:

• reduce warm cache launch latency
• (maybe) add launch args caching, to reduce launch latency
• make dataclasses.dataclass nestable

Planned pruning:

• remove argpack
• remove ti CLI
• remove support for NVidia GPUs earlier than sm_60/Pascal

What is gstaichi?

GsTaichi is a high performance multi-platform compiler, targeted at physics simulations. It compiles Python code into parallelizable kernels that can run on:

• NVidia GPUs, using CUDA
• Vulkan-compatible GPUs, using SPIR-V
• Mac Metal GPUs
• x86 and arm64 CPUs

GsTaichi supports automatic differentiation. GsTaichi lets you build fully fused GPU kernels, using Python.

Genesis simulator's best-in-class performance can be largely attributed to Taichi, its underlying GPU acceleration framework for Python. Given how critical is this component, we decided to fork Taichi and build our own very framework from there, so that from now on, we are free to drive its development in the direction that best supports the continuous improvement of Genesis simulator.

Installation

Prerequisites

• Python 3.10-3.13
• Mac OS 14, 15, Windows, or Ubuntu 22.04-24.04 or compatible

Procedure

pip install gstaichi

(For how to build from source, see our CI build scripts, e.g. linux build scripts )

Documentation

• docs
• API reference

Something is broken!

• Create an issue

Acknowledgements

• The original Taichi was developed with love by many contributors over many years. For the full list of contributors and credits, see Original taichi contributors