alphafold/relax/amber_minimize.py · 8f1ebd58960b7eed708cf5eebcc88a7e6a3bf8a6 · ModelZoo / alphafold2_jax · GitLab

Find file Blame History Permalink

Fix GPU relax for longer chains by pinning large memory ops to cpu. · 8f1ebd58
Ali Cowen-Rivers authored Jan 10, 2023
```
PiperOrigin-RevId: 501105389
Change-Id: I6c981d1d3231e008ebae192edb4586479eb5eb34
```
8f1ebd58

amber_minimize.py 18.6 KB