search

patrick-kidger / jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
Other
1.05k stars 49 forks source link

Fix `jax` dependency issue when `jaxlib` is not installed. #191

Closed ar0ck closed 4 months ago

ar0ck commented 4 months ago

See https://github.com/patrick-kidger/jaxtyping/issues/188#issuecomment-1984247889.

patrick-kidger commented 4 months ago

This LGTM! Thank you for the fix.