taichi 0.9.2

The Taichi Programming Language

---

import taichi as ti

What is Taichi?

Taichi is an open-source, imperative, parallel programming language for high-performance numerical computation. It is embedded in Python and uses just-in-time (JIT) compiler frameworks, for example LLVM, to offload the compute-intensive Python code to the native GPU or CPU instructions.

The language has broad applications spanning real-time physical simulation, numberical computation, augmented reality, artificial intelligence, vision and robotics, visual effects in films and games, general-purpose computing, and much more.

...More

Why Taichi?

• Built around Python: Taichi shares almost the same syntax with Python, allowing you to write algorithms with minimal language barrier. It is also well integrated into the Python ecosystem, including NumPy and PyTorch.
• Flexibility: Taichi provides a set of generic data containers known as SNode (/ˈsnoʊd/), an effective mechanism for composing hierarchical, multi-dimensional fields. This can cover many use patterns in numerical simulation (e.g. spatially sparse computing).
• Performance: With the @ti.kernel decorator, Taichi's JIT compiler automatically compiles your Python functions into efficient GPU or CPU machine code for parallel execution.
• Portability: Write your code once and run it everywhere. Currently, Taichi supports most mainstream GPU APIs, such as CUDA and Vulkan.
• ... and many more features! A cross-platform, Vulkan-based 3D visualizer, differentiable programming, quantized computation (experimental), etc.

Getting Started

Installation

Prerequisites

• Operating systems
  • Windows
  • Linux
  • macOS
• Python: 3.6 ~ 3.9 (64-bit only)
• Compute backends
  • x64/ARM CPUs
  • CUDA
  • Vulkan
  • OpenGL (4.3+)
  • Apple Metal
  • WebAssembly (experiemental)

Use Python's package installer pip to install Taichi:

pip install taichi

We also provide a nightly package. Note that nighly packages may crash because they are not fully tested. We cannot guarantee their validity, and you are at your own risk trying out our latest, untested features.

pip install -i https://test.pypi.org/simple/ taichi-nightly

Run your "Hello, world!"

Here is how you can program a 2D fractal in Taichi:

# python/taichi/examples/simulation/fractal.py

import taichi as ti

ti.init(arch=ti.gpu)

n = 320
pixels = ti.field(dtype=float, shape=(n * 2, n))


@ti.func
def complex_sqr(z):
    return ti.Vector([z[0]**2 - z[1]**2, z[1] * z[0] * 2])


@ti.kernel
def paint(t: float):
    for i, j in pixels:  # Parallelized over all pixels
        c = ti.Vector([-0.8, ti.cos(t) * 0.2])
        z = ti.Vector([i / n - 1, j / n - 0.5]) * 2
        iterations = 0
        while z.norm() < 20 and iterations < 50:
            z = complex_sqr(z) + c
            iterations += 1
        pixels[i, j] = 1 - iterations * 0.02


gui = ti.GUI("Julia Set", res=(n * 2, n))

for i in range(1000000):
    paint(i * 0.03)
    gui.set_image(pixels)
    gui.show()

If Taichi is properly installed, you should get the animation below 🎉:

See Get started for more information.

Build from source

If you wish to try our our experimental features or build Taichi for your own environments, see Developer installation.

Documentation

• Technical documents
• API Reference
• Blog

Contributing

Kudos to all of our amazing contributors! Taichi thrives through open-source. In that spirit, we welcome all kinds of contributions from the community. If you would like to participate, check out the Contribution Guidelines first.

Contributor avatars are randomly shuffled.

Community

Join our discussions

• GitHub Discussions
• Taichi 中文论坛

Report an issue

• If you spot an technical or documentation issue, file an issue at GitHub Issues
• If you spot any security issue, mail directly to [e-mail].

Contact us

You can also join our community from Slack or WeChat. Drop us a message at [e-mail] first, and we'll follow up.

Reference

Demos

• Taichi examples
• Advanced Taichi examples
• DiffTaichi
• Taichi elements
• Taichi Houdini
• More...

Lectures & talks

• SIGGRAPH 2020 course on Taichi basics: YouTube, Bilibili, slides (pdf).
• Chinagraph 2020 用太极编写物理引擎: 哔哩哔哩
• GAMES 201 高级物理引擎实战指南 2020: 课件
• 太极图形课第一季：课件
• TaichiCon: Taichi Developer Conferences
• More to come...

Citations

If you use Taichi in your research, please cite the corresponding papers:

• (SIGGRAPH Asia 2019) Taichi: High-Performance Computation on Sparse Data Structures [Video] [BibTex] [Code]
• (ICLR 2020) DiffTaichi: Differentiable Programming for Physical Simulation [Video] [BibTex] [Code]
• (SIGGRAPH 2021) QuanTaichi: A Compiler for Quantized Simulations [Video] [BibTex] [Code]