PyTorch interface? (not an issue)

taichi-dev / difftaichi

10 differentiable physical simulators built with Taichi differentiable programming (DiffTaichi, ICLR 2020)

2.47k stars 260 forks source link

PyTorch interface? (not an issue) #12

Closed JaroslavVitku closed 4 years ago

JaroslavVitku commented 4 years ago

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 4 years ago

Hi Jarda, please find sample PyTorch interfaces here: https://github.com/taichi-dev/taichi/blob/master/tests/python/test_torch_ad.py

JaroslavVitku commented 4 years ago

Great, thank you!