distrax: migrate from deprecated jax.linear_util to jax.extend.linear_util - Githubissues

google-deepmind / distrax

Apache License 2.0

529 stars 32 forks source link

distrax: migrate from deprecated jax.linear_util to jax.extend.linear_util #262

Closed copybara-service[bot] closed 10 months ago

copybara-service[bot] commented 10 months ago

distrax: migrate from deprecated jax.linear_util to jax.extend.linear_util