flax
fdfc0bf4 - Merge pull request #4699 from google:nnx-jitwrapped

Commit

1 year ago

Merge pull request #4699 from google:nnx-jitwrapped PiperOrigin-RevId: 753372576

References

#4699 - [nnx] add JitWrapped API

Author

a-googler

a-googler

Parents

Loading