Skip to content

Commit 5146d16

Browse files
Added documentation
1 parent 4e5d671 commit 5146d16

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

docs/source/advanced_usage/trainingmodel.rst

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,22 @@ while maintaining the same prediction accuracy.
6363
Currently, these options are disabled by default as they are still being tested
6464
extensively by the MALA team in production. **Yet, activating them is highly recommended!**
6565

66+
Apple Silicon GPU
67+
~~~~~~~~~~~~~~~~~
68+
69+
If you are using a MacOS system on Apple Silicon, you can use the GPU cores
70+
therein with
71+
72+
.. code-block:: python
73+
74+
parameters = mala.Parameters()
75+
parameters.use_mps = True
76+
77+
MPS stands for the Metal Performance Shaders framework, the backend Pytorch
78+
employs to utilize the Apple Silicon GPU cores. Please note that this option
79+
only affects training performance, the LAMMPS code acceleration only works with
80+
CUDA GPUs for now.
81+
6682
Advanced training metrics
6783
****************************
6884

0 commit comments

Comments
 (0)