flax
f82b1535 - Merge pull request #3809 from google:nnx-all-jit-kwargs

Commit

2 years ago

Merge pull request #3809 from google:nnx-all-jit-kwargs PiperOrigin-RevId: 621087100

References

Author

a-googler

Parents