Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jun 10, 2024

• [JAX] Update users of jax.tree.map() to be more careful about how they handle Nones.

  

  hawkinsp

  authored and

  Copybara-Service

  committedJun 10, 2024
  a7b7e73

Commits on Jun 7, 2024

• make haiku work with upcoming JAX change to tree_map (being more careful about

  

  mattjj

  authored and

  Copybara-Service

  committedJun 7, 2024
  9ccaae1

Commits on May 22, 2024

• Make flax an optional dependency.

  

  tomhennigan

  authored and

  Copybara-Service

  committedMay 22, 2024
  0451621

Commits on May 15, 2024

• Replace deprecated jax.tree_* functions with jax.tree.*

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedMay 15, 2024
  bdaabf9

• Bump required JAX and TF versions to latest.

  

  tomhennigan

  authored and

  Copybara-Service

  committedMay 15, 2024
  84cdbbb

Commits on Apr 9, 2024

• Silence some pytype errors.

  

  rchen152

  authored and

  Copybara-Service

  committedApr 9, 2024
  48dcadd

Commits on Apr 8, 2024

• Replace deprecated jax.random.shuffle with jax.random.permutation

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedApr 8, 2024
  34a4682

Commits on Mar 11, 2024

• If I am understanding correctly, one line of the model code of

  

  philipmlong

  authored and

  Copybara-Service

  committedMar 11, 2024
  7324464

Commits on Mar 9, 2024

• Fix some JAX key reuse issues in haiku tests

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedMar 9, 2024
  30527da

Commits on Feb 28, 2024

• Cut 0.0.13.dev release.

  

  tomhennigan

  authored and

  Copybara-Service

  committedFeb 28, 2024
  ea820a6

• Cut 0.0.12 release.

  

  tomhennigan

  authored and

  Copybara-Service

  committedFeb 28, 2024
  59109e2

Commits on Feb 22, 2024

• Populate _auto_repr before running module constructor.

  

  tomhennigan

  authored and

  Copybara-Service

  committedFeb 22, 2024
  5c0f1e2

Commits on Feb 16, 2024

• Deactivate failsafe in hk.flax.lift to avoid false-positives.

  

  Haiku Contributor

  authored and

  Copybara-Service

  committedFeb 16, 2024
  8dc61b3

• Use tf-nightly for Haiku testing.

  

  tomhennigan

  authored and

  Copybara-Service

  committedFeb 16, 2024
  011fbad

Commits on Jan 29, 2024

• enable state with layer stack

  

  Haiku Contributor

  authored and

  Copybara-Service

  committedJan 29, 2024
  134763d

Commits on Jan 26, 2024

• modify layer_stack transparency map

  

  Haiku Contributor

  authored and

  Copybara-Service

  committedJan 26, 2024
  9be7c9a

Commits on Jan 18, 2024

• Skip broken test.

  

  tkoeppe

  authored and

  Copybara-Service

  committedJan 18, 2024
  0898b7b

Commits on Jan 17, 2024

• Haiku: fix jaxpr_info_test for jax v0.4.24

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedJan 17, 2024
  5404939

Commits on Nov 30, 2023

• Internal change

  

  Haiku Contributor

  authored and

  Copybara-Service

  committedNov 30, 2023
  6339353

Commits on Nov 17, 2023

• fix test not to depend on details of JAX shape error type/message

  

  mattjj

  authored and

  Copybara-Service

  committedNov 17, 2023
  af9497f

Commits on Nov 15, 2023

• Add a comment to force_name to avoid overuse.

  

  hamzamerzic

  authored and

  Copybara-Service

  committedNov 15, 2023
  ce05309

Commits on Nov 10, 2023

• Cut 0.0.12.dev release.

  

  tomhennigan

  authored and

  Copybara-Service

  committedNov 10, 2023
  f1e4f5f

• Cut 0.0.11 release.

  

  tomhennigan

  authored and

  Copybara-Service

  committedNov 10, 2023
  7c15f61

Commits on Nov 3, 2023

• Allow for different parameters to be created when reuse=True.

  

  tomhennigan

  authored and

  Copybara-Service

  committedNov 3, 2023
  2915c88

Commits on Nov 1, 2023

• [jex] replace extend.random.PRNGImpl with extend.random.define_prng_impl

  

  froystig

  authored and

  Copybara-Service

  committedNov 1, 2023
  f1a2f8c

Commits on Oct 26, 2023

• migrate from deprecated jax.linear_util to jax.extend.linear_util

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedOct 26, 2023
  afd7c2b

Commits on Oct 23, 2023

• Updated the hk.mixed_precision.set_policy docstring

  

  superbobry

  authored and

  Copybara-Service

  committedOct 23, 2023
  402a701

• Do not run doctests as part of the docs configuration

  

  superbobry

  authored and

  Copybara-Service

  committedOct 23, 2023
  a4be276

Commits on Oct 22, 2023

• haiku: avoid calls to deprecated jax.random APIs

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedOct 22, 2023
  86a00ea

Commits on Oct 20, 2023

• Updated jax.config import

  

  superbobry

  authored and

  Copybara-Service

  committedOct 20, 2023
  7a6faba

• haiku: avoid deprecated jax.random.default_prng_impl()

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedOct 20, 2023
  bee3a15

• Merge pull request #746 from superbobry:main

  

  Copybara-Service

  committedOct 20, 2023
  ca3cebc

• Install requirements jointly

  superbobry

  committedOct 20, 2023
  1e3c653

Commits on Sep 15, 2023

• haiku: fully support JAX typed PRNG keys

  

  Jake VanderPlas

  authored and

  Copybara-Service

  committedSep 15, 2023
  94d1362

• Ran pyupgrade --py39-plus **/*.py

  

  superbobry

  authored and

  Copybara-Service

  committedSep 15, 2023
  cbcbbaa

Pagination

PreviousNext