Skip to content

Commit 1a4c22c

Browse files
authored
Add SO(3)/O(3)-averaging calculator wrappers (#85)
1 parent 40ca85c commit 1a4c22c

File tree

6 files changed

+1162
-23
lines changed

6 files changed

+1162
-23
lines changed

docs/src/engines/ase.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,10 @@ Supported model outputs
2323
:py:meth:`ase.Atoms.get_forces`, …);
2424
- arbitrary outputs can be computed for any :py:class:`ase.Atoms` using
2525
:py:meth:`MetatomicCalculator.run_model`;
26+
- for non-equivariant architectures like
27+
`PET <https://docs.metatensor.org/metatrain/latest/architectures/pet.html>`_,
28+
rotationally-averaged energies, forces, and stresses can be computed using
29+
:py:class:`metatomic.torch.ase_calculator.SymmetrizedCalculator`.
2630

2731
How to install the code
2832
^^^^^^^^^^^^^^^^^^^^^^^

docs/src/torch/reference/ase.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,7 @@ not just the energy, through the
1717
.. autoclass:: metatomic.torch.ase_calculator.MetatomicCalculator
1818
:show-inheritance:
1919
:members:
20+
21+
.. autoclass:: metatomic.torch.ase_calculator.SymmetrizedCalculator
22+
:show-inheritance:
23+
:members:

0 commit comments

Comments
 (0)