JAX 0.3.18 seems to have included some updates to the way array types are handled (via the introduction of jax.Array).
This seems to have broken some things for those of us running type checkers. Basic operations like some_array.shape now result in errors in both pyright and mypy:
Tab completion for common array operations like .shape, .reshape(), .astype(), etc also seems broken.
Is this a bug? If so is there a fix planned?
Thanks so much!!
What jax/jaxlib version are you using?jax v0.3.18, v0.3.19
Which accelerator(s) are you using?No response
Additional system infoNo response
NVIDIA GPU infoNo response
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3
