pyamrex
pyamrex copied to clipboard
Taichi examples
Taichi is a Python-like JIT-compiled language that provides support for multidimensional arrays, sparse data structures, and template metaprogramming for CUDA and (soon) ROCm: https://docs.taichi-lang.org/docs/hello_world
It would be nice to have usage examples of pyAMReX with Taichi.
Sounds good, you should be able to use __cuda_array_interface__ and soon DLPack #454 to couple to it.
Please do not hesitate to also add to_taichi helpers that lazy-import taichi only when the method is called.
neat, thanks!