Skip to content

Issues: google-deepmind/dm-haiku

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

74 Open 175 Closed
74 Open 175 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Integrating vmap with BatchNorm
#788 opened Jul 21, 2024 by reemabdelrazek30
hk.BatchNorm with jax.vmap
#786 opened Jul 2, 2024 by reemabdelrazek30
hk.switch does not work inside a hk.vmap function when hk.set_state is used
#785 opened Jul 2, 2024 by rsmath
1
Wrapping the init function inside jax.jit
#778 opened May 14, 2024 by ksmdnl
1
Issue with wheels including docs and examples folder
#752 opened Nov 1, 2023 by ethanluoyc
1
Direct Feedback Alignment
#750 opened Oct 24, 2023 by ferqui
Wrong gradients in a Haiku network
#741 opened Oct 17, 2023 by SNMS95
Modules got silently "reused" with hk.vmap
#740 opened Oct 8, 2023 by jjyyxx
2
How to export haiku network parameters into Pytorch network?
#738 opened Sep 21, 2023 by AlgebraLoveme
hk.Conv2DTranspose takes FOREVER to initialize and compile
#724 opened Sep 6, 2023 by sokrypton
1
Please publish requirements.txt fix to pip
#717 opened Aug 12, 2023 by aeneev
Efficiency difference in using jax.lax.fori_loop vs looping over identical layers?
#709 opened Jul 28, 2023 by hrbigelow
2
Reservoir Computing with Haiku
#705 opened Jul 26, 2023 by Wotaker
hk.custom_getter with_sharding_constraint
#666 opened Jun 5, 2023 by kavorite
rnn classifies mnist
#645 opened May 4, 2023 by never-to-never
3
DeprecationWarning transform_with_state because of jax.xla vs jax.interpreters.xla
#634 opened Apr 15, 2023 by joeryjoery
[Question] Is it perfectly okay to use hk.next_rng_key() inside of a hk.scan function? This takes place inside a call method of a haiku module.
#632 opened Apr 10, 2023 by EdanToledo
Haiku initialization
#606 opened Feb 22, 2023 by adhikarirsr
How to reinitialize the hidden states of RNNs?
#587 opened Dec 22, 2022 by qlan3
Correct way to integrate tf2jax output with a hk.Module
#582 opened Dec 11, 2022 by rdilip
1
Mypy error from next_rng_key type inconsistency with jax PRNGKeyArray
#578 opened Nov 30, 2022 by hylkedonker
23 September error : AttributeError: 'Config' object has no attribute 'jax_experimental_name_stack'
#576 opened Nov 23, 2022 by f3mf3m
3
Feature request: pretrained checkpoints for dm-haiku models
#571 opened Nov 17, 2022 by rdilip
inspect.getfullargspec in module constructor causes problems with function wrappers
#570 opened Nov 16, 2022 by jonkhler
Re-running the training produces very different result
#552 opened Nov 1, 2022 by osbm
ProTip! Type g i on any issue or pull request to go back to the issue listing page.