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
Prevent module creation pyright type errors
#702
NeilGirdhar
closed
1 year ago
2
Bump pygments from 2.7.4 to 2.15.0 in /docs
#701
dependabot[bot]
closed
1 year ago
1
Warning: hk.LayerNorm when used in transformer decoder causes violation of autoregressive property
#700
hrbigelow
closed
1 year ago
1
Don't install requirements-test in docs CI.
#699
copybara-service[bot]
closed
1 year ago
0
fix version issues in docs build.
#698
copybara-service[bot]
closed
1 year ago
0
Fix docs build.
#697
copybara-service[bot]
closed
1 year ago
0
Add support for dilation to `hk.DepthwiseConvND`.
#696
copybara-service[bot]
closed
1 year ago
1
Cut Haiku 0.0.11.dev release.
#695
copybara-service[bot]
closed
1 year ago
0
Cut Haiku 0.0.10 release.
#694
copybara-service[bot]
closed
1 year ago
0
Change to jax.interpreters.xla for JAX==0.4.14
#693
kmheckel
closed
1 year ago
3
Bump JAX requirements to fix OSS tests.
#692
copybara-service[bot]
closed
1 year ago
0
Change docstring for pooling to only document full-rank window_shape/strides.
#691
copybara-service[bot]
closed
1 year ago
0
Edit hints and comments mentioning `jax.scan()` to `jax.lax.scan()`
#690
paulbricman
closed
1 year ago
1
Bump scipy from 1.7.1 to 1.10.0 in /docs
#689
dependabot[bot]
closed
1 year ago
2
Support model examples
#688
llCurious
closed
1 year ago
7
Is there a way to load parameters from Flax model?
#687
llCurious
closed
1 year ago
2
Suggestion: alias `Transformed`(WithState) apply to __call__
#686
hylkedonker
closed
1 year ago
2
Update requirements-test.txt.
#685
copybara-service[bot]
closed
1 year ago
0
Bump tensorflow from 2.9.1 to 2.11.1
#684
dependabot[bot]
closed
1 year ago
1
Remove references to deprecated jax aliases
#683
copybara-service[bot]
closed
1 year ago
0
Update requirements.
#682
copybara-service[bot]
closed
1 year ago
0
Update RTD setup to python 3.11.
#681
copybara-service[bot]
closed
1 year ago
0
More fine-grained mixed-precision configuration
#680
llCurious
closed
1 year ago
2
MultiHeadAttention memory usage reduction via tiling
#679
maxconway
opened
1 year ago
0
Internal
#678
copybara-service[bot]
closed
1 year ago
1
Internal
#677
copybara-service[bot]
closed
1 year ago
1
Internal
#676
copybara-service[bot]
closed
1 year ago
1
Use of numpy for causal masking in transformer example
#675
smonsays
closed
1 year ago
2
Internal
#674
copybara-service[bot]
closed
1 year ago
1
Add Lena to the Haiku citation.
#673
copybara-service[bot]
closed
1 year ago
0
Fix setup.py packages to avoid installing "examples" and "docs"
#672
natsukium
opened
1 year ago
0
Changing input batch size causes `UnexpectedTracerError`
#671
ujjwalnur
closed
1 year ago
2
Feature request: declare_global_parameter / get_global_parameter
#670
hrbigelow
closed
1 year ago
6
Internal change.
#669
copybara-service[bot]
closed
1 year ago
0
Add jax.lax.map equivalent.
#668
copybara-service[bot]
closed
1 year ago
1
[NumPy] Fix uses of functions deprecated in NumPy 1.25.
#667
copybara-service[bot]
closed
1 year ago
0
hk.custom_getter with_sharding_constraint
#666
kavorite
opened
1 year ago
0
Add broadcast_dims to hk.dropout.
#665
copybara-service[bot]
closed
1 year ago
1
Allow transparent application of layer_stack.
#664
copybara-service[bot]
closed
1 year ago
1
Document Typo
#663
keep9oing
closed
1 year ago
0
Improve the transformer example somewhat.
#662
copybara-service[bot]
closed
1 year ago
0
Bump tornado from 6.1 to 6.3.2 in /docs
#661
dependabot[bot]
closed
1 year ago
1
Avoid moving np.ndarray inputs to device in BatchApply.
#660
copybara-service[bot]
closed
1 year ago
1
Allow generating split params from layer_stack.
#659
copybara-service[bot]
closed
1 year ago
1
Fix LayerStack + Haiku state error and add test.
#658
copybara-service[bot]
closed
1 year ago
0
Add a warning that layer_stack does not work with Haiku modules with state.
#657
copybara-service[bot]
closed
1 year ago
0
Remove hk.experimental.profiler_name_scopes config option.
#656
copybara-service[bot]
closed
1 year ago
0
Remove mentions of `hk.experimental.profiler_name_scopes` in Haiku.
#655
copybara-service[bot]
closed
1 year ago
0
Bump requests from 2.26.0 to 2.31.0 in /docs
#654
dependabot[bot]
closed
1 year ago
1
haiku.base: handle both new-style and old-style PRNG keys
#653
copybara-service[bot]
opened
1 year ago
0
Previous
Next