patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/ observed · 2026-08-28
Health v2 · maintenance only
95/100
- Activity 98
- Release rhythm 88
- Longevity 100
Flags: no_license
How is this computed?
round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10) — computed 2026-09-03. Adoption (stars, forks) is never an input.
- gap_med: 20
- age_days: 1532
- days_rel: 81
- days_push: 15
- n_releases_24m: 16
Adoption not part of the score
1859 stars · 94 forks observed · 2026-08-28
What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded
A Python library providing type annotations and runtime type-checking for the shape and dtype of arrays and tensors in JAX, PyTorch, NumPy, MLX, and TensorFlow. It works alongside runtime checkers like typeguard or beartype.
Use cases
- annotate tensor shapes and dtypes in function signatures
- runtime-check that arrays have expected shapes
- catch shape mismatch bugs in JAX or PyTorch code
- add shape annotations to NumPy array functions
- validate tensor dimensions in deep learning pipelines
When to choose
- you write numerical or deep learning code in Python and want shape/dtype safety
- you use JAX, PyTorch, NumPy, MLX, or TensorFlow and want typed array APIs
- you want annotations compatible with typeguard or beartype
When to avoid
- you need static type checking only without runtime verification
- your project does not involve array or tensor computations
- you need a framework-agnostic type system beyond array shapes
Facets
library · maturity active
type-system testing machine-learning machine-learning deep-learning developer-tools programming-languages python cross-platform jax pytorch numpy tensor-shapes dtype-checking runtime-type-checking python-typing annotations
2 sources
- readme: https://github.com/patrick-kidger/jaxtyping · fetched 2026-08-28 · 3387a9f32625
- registry_pypi: https://pypi.org/pypi/jaxtyping/json · fetched 2026-08-29 · 924a7f8d68ef
Member repositories
| Repository | Role | Health v2 |
|---|---|---|
| patrick-kidger/jaxtyping | main | 95 |
For agents
markdown · JSON · MCP: product_card(name="patrick-kidger/jaxtyping")
Data as of 2026-08-30T08:39:29.467469+00:00 · Report a problem