-
Notifications
You must be signed in to change notification settings - Fork 233
Pull requests: google-deepmind/dm-haiku
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Use short aliases for common tree operations in Haiku.
#799
opened Oct 16, 2024 by
copybara-service
bot
Loading…
Replace
jax.linear_util
-> jax.extend.linear_util
to suppress DeprecationWarning
s
#739
opened Oct 6, 2023 by
copybara-service
bot
Loading…
Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.
#722
opened Sep 1, 2023 by
copybara-service
bot
Loading…
Fix setup.py packages to avoid installing "examples" and "docs"
#672
opened Jun 19, 2023 by
natsukium
Loading…
haiku.base: handle both new-style and old-style PRNG keys
#653
opened May 22, 2023 by
copybara-service
bot
Loading…
[JAX] Add an option to control the out-of-bounds indexing behavior of Haiku embedding layers.
#493
opened Aug 5, 2022 by
copybara-service
bot
Loading…
Fix "output without random key" variable reference
#347
opened Mar 29, 2022 by
almostimplemented
Loading…
ProTip!
no:milestone will show everything without a milestone.