alphafold/model/modules.py · e51692c9aede6a82b3936a096aa1456f07dfa886 · ModelZoo / alphafold2_jax · GitLab

Find file Blame History Permalink

Change softmax to use where and float32. · e51692c9
Hamish Tomlinson authored Mar 27, 2023
```
PiperOrigin-RevId: 519675443
Change-Id: If87e6d16189ddcc03bb8435308d37f5919353107
```
e51692c9

modules.py 72.4 KB