Replace `jax.linear_util` -> `jax.extend.linear_util` to suppress `DeprecationWarning`s#739

Open

copybara-service[bot] wants to merge 1 commit intomainfrom test_571299183

+7-6

Commits on Oct 6, 2023

Replace jax.linear_util -> jax.extend.linear_util to suppress DeprecationWarnings

alimuldal
authored and
copybara-github
committed