Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Optimize EnergyShifter of TorchANI #38

Merged
merged 8 commits into from
Dec 20, 2021
Merged

Conversation

raimis
Copy link
Contributor

@raimis raimis commented Nov 12, 2021

This assume that a molecule is constant, so the energy contribution from the atomic self-energies can be pre-computed.

  • Implement TochANIEnergyShifter
  • Test
  • Documentation

@raimis raimis added this to Regular in Accelerated NNP in OpenMM via automation Nov 12, 2021
@raimis raimis self-assigned this Nov 12, 2021
@raimis raimis marked this pull request as draft November 12, 2021 17:18
@raimis raimis marked this pull request as ready for review November 12, 2021 18:26
@raimis raimis requested a review from peastman November 12, 2021 18:26
@raimis
Copy link
Contributor Author

raimis commented Nov 12, 2021

@peastman could you review?

@peastman
Copy link
Member

What's the reason for including this here? It ought to be a trivial operation. Does TorchANI somehow manage to implement it in an inefficient way?

@raimis
Copy link
Contributor Author

raimis commented Nov 15, 2021

This is based on the assumption that the molecule is constant. TorchANI doesn't support such optimizations.

@raimis raimis mentioned this pull request Nov 15, 2021
3 tasks
@peastman
Copy link
Member

Wouldn't it make more sense to contribute this to TorchANI then? It seems out of scope for this repository.

@raimis raimis mentioned this pull request Nov 17, 2021
3 tasks
@raimis raimis moved this from Regular to Blockers in Accelerated NNP in OpenMM Nov 23, 2021
@raimis
Copy link
Contributor Author

raimis commented Dec 16, 2021

@peastman can we proceed with this?

@peastman
Copy link
Member

Yes, go ahead.

@raimis raimis merged commit cacb4f8 into openmm:master Dec 20, 2021
Accelerated NNP in OpenMM automation moved this from Blockers to Done Dec 20, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants