trax icon indicating copy to clipboard operation
trax copied to clipboard

Published 20 hours ago • verified publisher icon google

Remove uses of deprecated jax.lax.tie_in

Open copybara-service[bot] opened this issue 1 year ago • 0 comments

Remove uses of deprecated jax.lax.tie_in

This has been a no-op since jax v0.2.0, and passes the second argument through unchanged. tie_in will be deprecated as of jax v0.4.24; see https://github.com/google/jax/pull/19413

copybara-service[bot] avatar Jan 29 '24 19:01 copybara-service[bot]