flax
df60f611 - nnx: add complex test with scan + batchnorm + dropout

Commit

2 years ago

nnx: add complex test with scan + batchnorm + dropout

References

#3567 - [nnx] Add complex test with scan + batchnorm + dropout

Author

cgarciae

cgarciae

Parents

Loading