Implementation is hosted here: https://bitbucket.org/qirong_zhu/taichi_public/