Does deepmd-kit have any plan to add ROCm support, like the way TensorFlow did?

deepmodeling / deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

https://docs.deepmodeling.com/projects/deepmd/

GNU Lesser General Public License v3.0

1.48k stars 508 forks source link

Closed ipe-zhangyz closed 3 years ago

njzjz commented 3 years ago

Implemented in #656.