flax
b12efcc0 - add support for initializing carry variables in scan

Commit

2 years ago

add support for initializing carry variables in scan

References

#3228 - Add support for initializing variables_carry in nn.scan

Author

cgarciae

cgarciae

Parents

Loading