NVIDIA / warp

A Python framework for high performance GPU simulation and graphics

Home Page:https://nvidia.github.io/warp/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Interoperability with pytorch geometric

nithinsomu opened this issue · comments

Just out of curiosity, does warp currently or in near future have plans to support interoperability with torch geometric? This would be useful when we have simulations in unstructured grid and want to use graph neural networks.

Hi @nithinsomu, I'm not particularly familiar with Torch Geometric - can you expand on which particular features would be the most useful? Or is it mostly a matter of data format conversions we should support?

Hey Miles,
Thank you for the reply. I was thinking along the lines of supporting the graph data format which is used in pytorch geometric, performing the transformations and may be some of the graph convolution operations if possible.

Thanks and Regards,
Nithin Somasekharan