Pull requests: google-deepmind/dm-haiku
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Fix _compute_fans when fan_in_axes is specified and len(shape) is 2.
#764
opened Jan 20, 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!
Find all pull requests that aren't related to any open issues with -linked:issue.