google-deepmind/dm-haiku

Issues

• +ดู~ดวงใจเทวพรหม พรชีวัน (EP.6) ตอนที่ 6 ดูย้อนหลังเต็ม เรื่อง ULTRA~HD ตอนล่าสุดฟรี

  #790 opened 4 months ago by aoopooyaa
  0

• Integrating vmap with BatchNorm

  #788 opened 4 months ago by reemabdelrazek30
  0

• hk.BatchNorm with jax.vmap

  #786 opened 5 months ago by reemabdelrazek30
  0

• hk.switch does not work inside a hk.vmap function when hk.set_state is used

  #785 opened 5 months ago by rsmath
  1

• Consider make flax an optional dependency

  #780 opened 6 months ago by jjyyxx
  1

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

  #778 opened 6 months ago by ksmdnl
  1

• Train multiple hk.nets.MLP with one optimizer

  #766 opened 7 months ago by rsmath
  2

• rnn classifies mnist

  #645 opened 2 years ago by never-to-never
  3

• TypeError: 'type' object is not subscriptable

  #774 opened 8 months ago by chenyzh28
  4

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

  #579 opened 2 years ago by noahadhikari
  2

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

  #755 opened a year ago by simon-bachhuber
  1

• 0.4.16 timeline

  #737 opened a year ago by joaogui1
  2

• Issue with wheels including docs and examples folder

  #752 opened a year ago by ethanluoyc
  0

• Direct Feedback Alignment

  #750 opened a year ago by ferqui
  0

• Wrong gradients in a Haiku network

  #741 opened a year ago by SNMS95
  0

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

  #740 opened a year ago by jjyyxx
  2

• How to export haiku network parameters into Pytorch network?

  #738 opened a year ago by AlgebraLoveme
  0

• How to use `apply` with additional parameters?

  #723 opened a year ago by Lando-L
  1

• Document Typo

  #663 opened a year ago by keep9oing
  0

• hk.Conv2DTranspose takes FOREVER to initialize and compile

  #724 opened a year ago by sokrypton
  1

• Please publish requirements.txt fix to pip

  #717 opened a year ago by aeneev
  0

• Use of numpy for causal masking in transformer example

  #675 opened a year ago by smonsays
  2

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

  #709 opened a year ago by hrbigelow
  2

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

  #687 opened a year ago by llCurious
  2

• Reservoir Computing with Haiku

  #705 opened a year ago by Wotaker
  0

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

  #700 opened a year 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

• Orthogonal initilizer does not support MLP?

  #635 opened 2 years ago by rezunli96
  1

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

  #634 opened 2 years 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 2 years ago by EdanToledo
  0

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

  #622 opened 2 years ago by cmunna0052
  2

• Analog to `flax.struct.dataclass`

  #610 opened 2 years ago by homerjed
  2

• Haiku initialization

  #606 opened 2 years ago by adhikarirsr
  0

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

  #588 opened 2 years ago by hylkedonker
  2

• How to use ConvLSTM?

  #584 opened 2 years ago by alembcke
  2

• Best Practice for using hk.next_rng_key()

  #585 opened 2 years ago by wbrenton
  2

• How to reinitialize the hidden states of RNNs?

  #587 opened 2 years ago by qlan3
  0

• [Minor bug] error check in hk.transform_with_state

  #583 opened 2 years ago by albertfgu
  1

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

  #582 opened 2 years 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

• Feature request: pretrained checkpoints for dm-haiku models

  #571 opened 2 years ago by rdilip
  0