Commits · a31061e8c3762e89eed14ac327a97c0b12e5fdeb · ModelZoo / alphafold2_jax

13 Oct, 2022 1 commit

[JAX] Import scipy submodules directly in code that uses them. · a31061e8

Peter Hawkins authored Oct 13, 2022

JAX will soon import scipy lazily, so code that relies on JAX having already imported submodules of scipy will fail. Change code to import what it uses.

PiperOrigin-RevId: 480841853
Change-Id: I18a744a561d73da3ccd43c26fcd99d10070c058e

a31061e8

31 May, 2022 1 commit
- Fix jax.tree_multimap deprecation warning. · d9e5e1d9
  Augustin Zidek authored May 31, 2022
```
PiperOrigin-RevId: 451994826
Change-Id: I4573baf61d33010c75de717d3b49f47bc9c6a8ac
```
  d9e5e1d9
27 Jul, 2021 1 commit

Fix import ordering. · 1e216f93

Tom Ward authored Jul 27, 2021

PiperOrigin-RevId: 387085679
Change-Id: I73287fcd0a29e899543b64c596e306195a2f435e

1e216f93

15 Jul, 2021 1 commit
- Initial release of AlphaFold. · 1109480e
  Augustin-Zidek authored Jul 15, 2021
```
PiperOrigin-RevId: 384954738
```
  1109480e