Skip to content

GitLab

Switch branch/tag
  1. 21 Feb, 2022 1 commit
  3. 18 Feb, 2022 1 commit
  5. 16 Feb, 2022 1 commit
    • Michael Figurnov's avatar
      Improve support for num_recycle=0. · 2cd61ade
      Michael Figurnov authored 
      Previously, setting num_recycle=0 in a pretrained recycling model skipped creating some Linears and LayerNorms. This meant that the the offsets of these modules were not applied leading to a degraded performance in that case.

PiperOrigin-RevId: 429075328
Change-Id: I9257f859521799f45e2deef3803c249311051225
      2cd61ade
  7. 14 Feb, 2022 1 commit
    • Tom Hennigan's avatar
      Add split_rng=False (current default) to sharded_map. · 929b188a
      Tom Hennigan authored 
      Haiku plans to make split_rng a required argument to hk.vmap in an upcoming
release. This change updates AlphaFold to preserve the current behaviour. We
also handle the case where users are using a release of Haiku without the
split_rng option, for these users split_rng=False is implied.

PiperOrigin-RevId: 428429472
Change-Id: I3292396eb330ffabf2f36f5364aa0ae1bc74cf71
      929b188a
  9. 08 Feb, 2022 1 commit
  11. 27 Jan, 2022 1 commit
  13. 04 Nov, 2021 1 commit
  15. 02 Nov, 2021 1 commit
  17. 10 Sep, 2021 1 commit
  19. 31 Aug, 2021 1 commit
  21. 02 Aug, 2021 1 commit
    • Augustin Zidek's avatar
      Fix a few typos. · 41232dc7
      Augustin Zidek authored 
      PiperOrigin-RevId: 388172326
Change-Id: I11d9e498c226cd752947feb51b7d1eb343b4d7ab
      41232dc7
  23. 30 Jul, 2021 1 commit
  25. 29 Jul, 2021 1 commit
  27. 27 Jul, 2021 1 commit
    • Tom Ward's avatar
      Fix import ordering. · 1e216f93
      Tom Ward authored 
      PiperOrigin-RevId: 387085679
Change-Id: I73287fcd0a29e899543b64c596e306195a2f435e
      1e216f93
  29. 26 Jul, 2021 1 commit
  31. 22 Jul, 2021 1 commit
  33. 15 Jul, 2021 1 commit