jax
fc2edbfa - Add a `freeze` primitive to delimit ref lifetimes for AD.

Commit

1 year ago

Add a `freeze` primitive to delimit ref lifetimes for AD. Also some basic AD through mutable_array/freeze. Co-authored-by: Matthew Johnson <mattjj@google.com>

References

#25369 - Add a `freeze` primitive to delimit ref lifetimes for AD.

#176 - CI: 12/10/24 upstream sync

Author

dougalm

Committer

dougalm

Parents

39d73a68

jax fc2edbfa - Add a `freeze` primitive to delimit ref lifetimes for AD.

jax
fc2edbfa - Add a `freeze` primitive to delimit ref lifetimes for AD.