google-deepmind/dm-haiku

[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

Title says it all. just a question as I'm worried about potential JIT compilation issues that might occur with haiku state etc.