Allow user to disable jax import

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

Closed MilesCranmer closed 4 months ago

MilesCranmer commented 4 months ago

Fixes #178

MilesCranmer commented 4 months ago

Fixed by ff3ded32885d1f6d41d796bb56ec4db656a04ee5 instead!