Repolode
提交项目
taichi-dev

taichi-dev/taichi

嵌入 Python 的编程语言，可将数值计算代码即时编译为 CPU 或 GPU（CUDA、Vulkan、Metal 等）上运行的高性能代码。

在 GitHub 查看官网
Star
28,450
Fork
2,392
未关闭 Issue
926
最近提交
上个月

这是什么？

Taichi Lang 是一门嵌入 Python 的命令式并行编程语言，面向高性能数值计算。使用 @ti.kernel 装饰的函数会由基于 LLVM 的 JIT 编译器编译为 CPU 或 GPU 的原生指令，无需改用 C++ 或 CUDA 也能并行运行计算密集型代码。

语法与 Python 基本一致，并可与 NumPy、PyTorch 配合使用。其 SNode 数据容器可以描述层级化、多维的场，包括空间稀疏的数据结构。同一份代码可在 CUDA、Vulkan、OpenGL、Apple Metal 以及 x64/ARM CPU 等后端上运行。

项目还支持可微分编程，并提供基于 Vulkan 的 3D 可视化工具。示例涵盖物理仿真、渲染和流体模拟等场景。

核心功能

  • 01使用接近 Python 的语法编写内核，并即时编译为 CPU 或 GPU 代码。
  • 02支持 CUDA、Vulkan、OpenGL、Apple Metal 以及 x64/ARM CPU。
  • 03通过 SNode 容器支持空间稀疏的数据结构。
  • 04可与 NumPy 和 PyTorch 集成。
  • 05包含可微分编程功能和基于 Vulkan 的 3D 可视化工具。

适用场景

  • 在 Python 中加速流体、粒子等物理数值仿真。
  • 无需手写 CUDA 或着色器代码，快速验证 GPU 算法。
  • 构建可微分的物理或图形计算流程。
  • 开发需要并行计算的渲染或视觉特效实验。

快速上手

  1. 1

    使用 pip 安装 Taichi

    pip install taichi
  2. 2

    启动示例画廊

    ti gallery

使用前须知

  • 仅支持 Python 3.6 至 3.10 的 64 位版本。
  • 夜间构建版本未经充分测试，可能崩溃。

收录于专题

GPU 加速的 AI 训练与推理工具机器人与具身智能项目编译器、转译器与编程语言

标签

#gpu-programming#python#high-performance-computing#simulation#compiler#differentiable-programming#computer-graphics#gpu#sparse-computation#taichi

常见问题

taichi 是什么？
嵌入 Python 的编程语言，可将数值计算代码即时编译为 CPU 或 GPU（CUDA、Vulkan、Metal 等）上运行的高性能代码。 Taichi Lang 是一门嵌入 Python 的命令式并行编程语言，面向高性能数值计算。使用 @ti.kernel 装饰的函数会由基于 LLVM 的 JIT 编译器编译为 CPU 或 GPU 的原生指令，无需改用 C++ 或 CUDA 也能并行运行计算密集型代码。
taichi 可以免费使用吗？用的什么许可证？
taichi 以 Apache-2.0 许可证开源。用于商业产品前，请先确认许可证条款。
taichi 还在维护吗？
半年内有更新。taichi 最近一次提交是在 2026年10月5日。
taichi 适合用来做什么？
在 Python 中加速流体、粒子等物理数值仿真。 无需手写 CUDA 或着色器代码，快速验证 GPU 算法。 构建可微分的物理或图形计算流程。 开发需要并行计算的渲染或视觉特效实验。
taichi 怎么安装和运行？
使用 pip 安装 Taichi: pip install taichi 启动示例画廊: ti gallery
在哪里获取 taichi？
源代码和安装说明在 GitHub 上：https://github.com/taichi-dev/taichi

概要由 AI 根据项目 README 生成；GitHub 数据每日更新。

相关项目