torchmd-net icon indicating copy to clipboard operation
torchmd-net copied to clipboard

Published 20 hours ago • verified publisher icon torchmd

POC: the graph network optimization with the custom kernels

Open raimis opened this issue 2 years ago • 0 comments

  • [x] Integrate the neighbor kernel
  • [x] Integrate the message kernel
  • [x] Benchmark

Depend on #61 and #69

raimis avatar Apr 20 '22 09:04 raimis