See also jaxtyping which, contrary to what its name might imply, covers JAX/PyTo... | Hacker News

Hacker Newsnew | past | comments | ask | show | jobs | submit

nestorD 5 months ago | parent | context | favorite | on: Futhark by example (2020)

See also jaxtyping which, contrary to what its name might imply, covers JAX/PyTorch/NumPy/MLX/TensorFlow arrays and tensors.

https://docs.kidger.site/jaxtyping/

ainch 5 months ago [–]

I use jaxtyping as documentation, but the fact it can only be used for runtime checking (in a slightly clunky manner) and can't infer shapes based on ops really limits its utility imo.

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact