Skip to content

Add Symplectic FlashMD Model#1010

Open
johannes-spies wants to merge 45 commits intomainfrom
symplectic
Open

Add Symplectic FlashMD Model#1010
johannes-spies wants to merge 45 commits intomainfrom
symplectic

Conversation

@johannes-spies
Copy link
Contributor

@johannes-spies johannes-spies commented Jan 15, 2026

Add support for training a symplectic version of FlashMD.


📚 Documentation preview 📚: https://metatrain--1010.org.readthedocs.build/en/1010/

@johannes-spies
Copy link
Contributor Author

cscs-ci run

@johannes-spies
Copy link
Contributor Author

Yes, a new architecture is the best way to go because there is a severe difference in the forward pass (autograd over the S3 action) that needs to be expressed. I definitely agree with you on the code duplication. To clean it up, I will import modules from FlashMD as much as possible!

Copy link
Collaborator

@frostedoyster frostedoyster left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

There are a few things to fix, but they should all be relatively effortless

Copy link
Collaborator

@frostedoyster frostedoyster left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry, extra comment

@frostedoyster
Copy link
Collaborator

Also, can we have log_mae on by default?

@johannes-spies
Copy link
Contributor Author

cscs-ci run

1 similar comment
@johannes-spies
Copy link
Contributor Author

cscs-ci run

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants