google-deepmind/dm-haiku

Issues

• Consider make flax an optional dependency

  #780 opened 11 days ago by jjyyxx
  1

• Wrapping the ```init``` function inside ```jax.jit```

  #778 opened 19 days ago by ksmdnl
  1

• Train multiple hk.nets.MLP with one optimizer

  #766 opened 2 months ago by rsmath
  2

• rnn classifies mnist

  #645 opened a year ago by never-to-never
  3

• TypeError: 'type' object is not subscriptable

  #774 opened 2 months ago by chenyzh28
  4

• BatchNorm No value for 'average' ValueError, perhaps set an init function?

  #579 opened a year ago by noahadhikari
  2

• `haiku.experimental.flax` is not part of newest pip release

  #755 opened 7 months ago by SimiPixel
  1

• 0.4.16 timeline

  #737 opened 7 months ago by joaogui1
  2

• Issue with wheels including docs and examples folder

  #752 opened 7 months ago by ethanluoyc
  0

• Direct Feedback Alignment

  #750 opened 7 months ago by ferqui
  0

• Wrong gradients in a Haiku network

  #741 opened 8 months ago by SNMS95
  0

• Modules got silently "reused" with `hk.vmap`

  #740 opened 8 months ago by jjyyxx
  2

• How to export haiku network parameters into Pytorch network?

  #738 opened 8 months ago by AlgebraLoveme
  0

• How to use `apply` with additional parameters?

  #723 opened 9 months ago by Lando-L
  1

• Document Typo

  #663 opened 9 months ago by keep9oing
  0

• hk.Conv2DTranspose takes FOREVER to initialize and compile

  #724 opened 9 months ago by sokrypton
  1

• Please publish requirements.txt fix to pip

  #717 opened 10 months ago by aeneev
  0

• Use of numpy for causal masking in transformer example

  #675 opened 10 months ago by smonsays
  2

• Efficiency difference in using jax.lax.fori_loop vs looping over identical layers?

  #709 opened 10 months ago by hrbigelow
  2

• Is there a way to load parameters from Flax model?

  #687 opened 10 months ago by llCurious
  2

• Reservoir Computing with Haiku

  #705 opened 10 months ago by Wotaker
  0

• Warning: hk.LayerNorm when used in transformer decoder causes violation of autoregressive property

  #700 opened 10 months ago by hrbigelow
  1

• Change to jax.interpreters.xla for JAX==0.4.14

  #693 opened a year ago by kmheckel
  3

• Support model examples

  #688 opened a year ago by llCurious
  7

• More fine-grained mixed-precision configuration

  #680 opened a year ago by llCurious
  2

• Suggestion: alias `Transformed`(WithState) apply to __call__

  #686 opened a year ago by hylkedonker
  2

• Improving compilation speed for repeated layers

  #639 opened a year ago by davisyoshida
  0

• Feature request: declare_global_parameter / get_global_parameter

  #670 opened a year ago by hrbigelow
  6

• Changing input batch size causes `UnexpectedTracerError`

  #671 opened a year ago by ujjwalnur
  2

• hk.custom_getter with_sharding_constraint

  #666 opened a year ago by kavorite
  0

• vmap and lift_with_state

  #557 opened a year ago by eserie
  0

• Orthogonal initilizer does not support MLP?

  #635 opened a year ago by rezunli96
  1

• DeprecationWarning `transform_with_state` because of `jax.xla` vs `jax.interpreters.xla`

  #634 opened a year ago by joeryjoery
  0

• [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 a year ago by EdanToledo
  0

• Is it impossible to turn a sequence of identical instance blocks into a compiled loop?

  #622 opened a year ago by cmunna0052
  2

• Analog to `flax.struct.dataclass`

  #610 opened a year ago by homerjed
  2

• Haiku initialization

  #606 opened a year ago by adhikarirsr
  0

• Argument `init` in `get_parameter` is not optional

  #588 opened a year ago by hylkedonker
  2

• How to use ConvLSTM?

  #584 opened a year ago by alembcke
  2

• Best Practice for using hk.next_rng_key()

  #585 opened a year ago by wbrenton
  2

• How to reinitialize the hidden states of RNNs?

  #587 opened a year ago by qlan3
  0

• [Minor bug] error check in hk.transform_with_state

  #583 opened a year ago by albertfgu
  1

• Correct way to integrate tf2jax output with a hk.Module

  #582 opened a year ago by rdilip
  1

• 23 September error : AttributeError: 'Config' object has no attribute 'jax_experimental_name_stack'

  #576 opened 2 years ago by f3mf3m
  3

• Mypy error from `next_rng_key` type inconsistency with jax `PRNGKeyArray`

  #578 opened 2 years ago by hylkedonker
  0

• VQ-VAE Notebook Example

  #577 opened 2 years ago by wbrenton
  0

• `jax.config.jax_experimental_name_stack` removed in JAX v0.3.25

  #565 opened 2 years ago by alembcke
  4

• Haiku equivalent to Keras `Lambda`

  #562 opened 2 years ago by alembcke
  2

• Feature request: pretrained checkpoints for dm-haiku models

  #571 opened 2 years ago by rdilip
  0

• inspect.getfullargspec in module constructor causes problems with function wrappers

  #570 opened 2 years ago by jonkhler
  0