/triton-torchgeo-gat-example

An example of running a pytorch-geometric graph attention model in nvidia triton.

Primary LanguagePythonMIT LicenseMIT

Stargazers