flax
d111adf7 - Merge pull request #4191 from google:nnx-optimize-jit

Commit

1 year ago

Merge pull request #4191 from google:nnx-optimize-jit PiperOrigin-RevId: 674457375

References

Author

a-googler

Parents