search

openmm / openmm-ml

High level API for using machine learning models in OpenMM simulations
Other
76 stars 25 forks source link

Register species and pbc as parameters #28

Closed peastman closed 2 years ago

peastman commented 2 years ago

Fixes #27. If they aren't registered as parameters, then calling to(device) on the module doesn't move them to the correct device.