1. 21 Feb, 2022 1 commit
  2. 18 Feb, 2022 2 commits
  3. 17 Feb, 2022 1 commit
  4. 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
  5. 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
  6. 08 Feb, 2022 1 commit
  7. 31 Jan, 2022 1 commit
  8. 28 Jan, 2022 1 commit
  9. 27 Jan, 2022 1 commit
  10. 19 Jan, 2022 2 commits
  11. 05 Nov, 2021 2 commits
  12. 04 Nov, 2021 1 commit
  13. 02 Nov, 2021 1 commit
  14. 10 Sep, 2021 2 commits
  15. 31 Aug, 2021 2 commits
  16. 25 Aug, 2021 1 commit
  17. 13 Aug, 2021 3 commits
  18. 12 Aug, 2021 1 commit
  19. 02 Aug, 2021 2 commits
  20. 30 Jul, 2021 2 commits
  21. 29 Jul, 2021 3 commits
  22. 27 Jul, 2021 3 commits
  23. 26 Jul, 2021 2 commits
  24. 23 Jul, 2021 1 commit
    • DeepMind's avatar
      Fix a link in the README · d3a4eefd
      DeepMind authored
      PiperOrigin-RevId: 386417647
      Change-Id: I6f2121f3c02d3ef97467f46c5e651ca1b0f2eab2
      d3a4eefd
  25. 22 Jul, 2021 1 commit
  26. 16 Jul, 2021 1 commit