flax
595e711f - Merge pull request #4472 from google:nnx-fix-fori

Commit

1 year ago

Merge pull request #4472 from google:nnx-fix-fori PiperOrigin-RevId: 712665956

References

#4472 - Fix main and add nnx.fori_loop test

Author

a-googler

a-googler

Parents

Loading