docs(backport): Update JAX backend docstring types to jax.Array#2152

Merged

matthewfeickert merged 1 commit intorelease/v0.7.xfrom backport/jax-docstrings

Mar 31, 2023

+30-31

Commits on Mar 31, 2023

docs(backport): Update JAX backend docstring types to jax.Array
matthewfeickert
committed