File tree Expand file tree Collapse file tree
docs/source/advanced_usage Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -63,6 +63,22 @@ while maintaining the same prediction accuracy.
6363Currently, these options are disabled by default as they are still being tested
6464extensively 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+
6682Advanced training metrics
6783****************************
6884
You can’t perform that action at this time.
0 commit comments