PyTorch interface? (not an issue)
JaroslavVitku opened this issue · 2 comments
JaroslavVitku commented
Hi, I've found your simulator and it looks really nice.
I wanted to ask whether there is possibility to interface the environment (including gradients) with PyTorch. And if not, how complicated this might be to do on my own?
Best,
Jarda
yuanming-hu commented
Hi Jarda, please find sample PyTorch interfaces here: https://github.com/taichi-dev/taichi/blob/master/tests/python/test_torch_ad.py
JaroslavVitku commented
Great, thank you!