flax
b19772eb - fix rng position, rng carry stream, scan arg

Commit

2 years ago

fix rng position, rng carry stream, scan arg

References

#4272 - Porting RNN from Linen to NNX

Author

zinccat

zinccat

Parents

Loading