Productive, portable, and performant GPU programming in Python.
The Taichi Programming Language
Leveraging Taichi Lang to customize brain dynamics operators.
Up to 913x faster Lie algebra products for all five exceptional algebras with differentiable PyTorch equivariant layers