jax
182c4622 - hoist constants to parameters in jit-compiled functions

Commit

5 years ago

hoist constants to parameters in jit-compiled functions Co-authored-by: Matthew Johnson <mattjj@google.com>

References

Author

froystig

Parents