Remove references to deprecated jax.ShapedArray

google / trax

Trax — Deep Learning with Clear Code and Speed

Apache License 2.0

8.01k stars 813 forks source link

Remove references to deprecated jax.ShapedArray #1775

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Remove references to deprecated jax.ShapedArray

This is deprecated as of https://github.com/google/jax/pull/15263: most users will never need to use ShapedArray directly, and so having it exposed in the top-level public namespace causes undue confusion.