fix: Use `jax.errors` API as `jax._src` removed from public namespace#2034

Merged

matthewfeickert merged 1 commit intomainfrom fix/jax-_src

Sep 26, 2022

+2-2

Commits on Sep 26, 2022

fix: Use jax.errors as jax 0.3.18 removes import of jax._src
matthewfeickert
committed