[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.
EdanToledo opened this issue · 0 comments
EdanToledo commented
Title says it all. just a question as I'm worried about potential JIT compilation issues that might occur with haiku state etc.