jax
1b6e0981 - Only consider Tracers with ShapedArray avals as instances of Array.

Commit

94 days ago

Only consider Tracers with ShapedArray avals as instances of Array. This fixes a TODO from https://github.com/jax-ml/jax/pull/33420. PiperOrigin-RevId: 861851484

References

#33530 - Only consider Tracers with ShapedArray avals as instances of Array.

Author

hawkinsp

Committer

Google-ML-Automation

Parents

9f327f69

jax 1b6e0981 - Only consider Tracers with ShapedArray avals as instances of Array.

jax
1b6e0981 - Only consider Tracers with ShapedArray avals as instances of Array.