This repository was archived by the owner on Apr 24, 2024. It is now read-only.
This repository was archived by the owner on Apr 24, 2024. It is now read-only.
Example using ModuleTensorMap #65
Description
When #62 is merged, we need an example using the ModuleTensorMap to show how to create custom models. I was thinking something very rudimentary for now: One nonlinear layer and then a reduce operation to sum over all contributions to get a total energy. More complicated things can be done in a dedicated modelssubmodule.
Metadata
Metadata
Assignees
Labels
No labels