issues
search
google-deepmind
/
dm-haiku
JAX-based neural network library
https://dm-haiku.readthedocs.io
Apache License 2.0
2.91k
stars
231
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Issue with wheels including docs and examples folder
#752
ethanluoyc
opened
1 year ago
0
migrate from deprecated jax.linear_util to jax.extend.linear_util
#751
copybara-service[bot]
closed
1 year ago
0
Direct Feedback Alignment
#750
ferqui
opened
1 year ago
0
Do not run doctests as part of the docs configuration
#749
copybara-service[bot]
closed
1 year ago
0
Updated the `hk.mixed_precision.set_policy` docstring
#748
copybara-service[bot]
closed
1 year ago
0
haiku: avoid calls to deprecated jax.random APIs
#747
copybara-service[bot]
closed
1 year ago
0
Install requirements jointly
#746
superbobry
closed
1 year ago
0
[jex] replace `extend.random.PRNGImpl` with `extend.random.define_prng_impl`
#745
copybara-service[bot]
closed
1 year ago
0
haiku: avoid deprecated jax.random.default_prng_impl()
#744
copybara-service[bot]
closed
1 year ago
0
Updated jax.config import
#743
copybara-service[bot]
closed
1 year ago
0
Updated jax.config import
#742
copybara-service[bot]
closed
1 year ago
0
Wrong gradients in a Haiku network
#741
SNMS95
opened
1 year ago
0
Modules got silently "reused" with `hk.vmap`
#740
jjyyxx
opened
1 year ago
2
Replace `jax.linear_util` -> `jax.extend.linear_util` to suppress `DeprecationWarning`s
#739
copybara-service[bot]
opened
1 year ago
0
How to export haiku network parameters into Pytorch network?
#738
AlgebraLoveme
opened
1 year ago
0
0.4.16 timeline
#737
joaogui1
closed
1 year ago
2
haiku: fully support JAX typed PRNG keys
#736
copybara-service[bot]
closed
1 year ago
0
[haiku] Fix deprecated usage of `linear_util` in `dot.py`.
#735
copybara-service[bot]
closed
1 year ago
0
Add error message to the layer stack assertion.
#734
copybara-service[bot]
closed
1 year ago
1
Fixed a typo in the parameter sharing notebook
#733
copybara-service[bot]
closed
1 year ago
0
Ran pyupgrade --py39-plus **/*.py
#732
copybara-service[bot]
closed
1 year ago
0
Run integration tests on 3.9-3.11
#731
copybara-service[bot]
closed
1 year ago
0
Avoid installing test dependencies when building docs on GitHub Actions
#730
copybara-service[bot]
closed
1 year ago
0
Do not test on 3.8, since we no longer support it
#729
copybara-service[bot]
closed
1 year ago
0
Avoid crashes when constructing signatures
#728
copybara-service[bot]
closed
1 year ago
0
Do not use ... as a type
#727
copybara-service[bot]
closed
1 year ago
0
ModuleMetaclass now directly subclasses abc.ABCMeta
#726
copybara-service[bot]
closed
1 year ago
0
Dropped 3.8 from the list of supported versions
#725
copybara-service[bot]
closed
1 year ago
0
hk.Conv2DTranspose takes FOREVER to initialize and compile
#724
sokrypton
opened
1 year ago
1
How to use `apply` with additional parameters?
#723
Lando-L
closed
1 year ago
1
Replace deprecated jax.linear_util.wrap_init with jax.extend.linear_util.wrap_init.
#722
copybara-service[bot]
opened
1 year ago
0
Removed the unused ``replaceable`` decorator
#721
copybara-service[bot]
closed
1 year ago
0
Replace uses of `jnp.array` in types with `jnp.ndarray`.
#720
copybara-service[bot]
closed
1 year ago
0
Bump tornado from 6.1 to 6.3.3 in /docs
#719
dependabot[bot]
closed
1 year ago
1
Use the same device for splitting keys.
#718
copybara-service[bot]
closed
1 year ago
0
Please publish requirements.txt fix to pip
#717
aeneev
opened
1 year ago
0
Fix Pyright error when creating modules
#716
NeilGirdhar
closed
1 year ago
0
Add utilities for working with a mixture of Haiku and Flax modules.
#715
copybara-service[bot]
closed
1 year ago
0
Graduate a number of stable features out of hk.experimental.
#714
copybara-service[bot]
closed
1 year ago
0
Fix typo
#713
copybara-service[bot]
closed
1 year ago
1
Recommend Flax in the README.
#712
copybara-service[bot]
closed
1 year ago
0
Internal Code Change
#711
copybara-service[bot]
closed
1 year ago
0
Fixed a latent type error in create_module_from_qualified_name
#710
copybara-service[bot]
closed
1 year ago
0
Efficiency difference in using jax.lax.fori_loop vs looping over identical layers?
#709
hrbigelow
opened
1 year ago
2
Minor code simplification for `_LayerStackNoPerLayer`.
#708
copybara-service[bot]
closed
1 year ago
0
modify layer_stack transparency map
#707
copybara-service[bot]
opened
1 year ago
1
Don't swallow errors with RNG keys in parameter initialisers.
#706
copybara-service[bot]
closed
1 year ago
0
Reservoir Computing with Haiku
#705
Wotaker
opened
1 year ago
0
Bump certifi from 2021.10.8 to 2023.7.22 in /docs
#704
dependabot[bot]
closed
1 year ago
1
Add .editorconfig
#703
NeilGirdhar
closed
1 year ago
0
Previous
Next