patrick-kidger/jaxtyping

Some data may be outdated — see provenance labels.

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/ observed · 2026-09-24

github.com/patrick-kidger/jaxtyping · Python · NOASSERTION (other) observed · 2026-09-24

Health v2 · maintenance only

90/100

Activity 93
Release rhythm 81
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-10-21. Adoption (stars, forks) is never an input.

gap_med: 20
age_days: 1580
days_rel: 129
days_push: 47
n_releases_24m: 16

Full methodology

Adoption not part of the score

1873 stars · 96 forks observed · 2026-09-24

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-09-24 · 3387a9f32625
registry_pypi: https://pypi.org/pypi/jaxtyping/json · fetched 2026-08-29 · 924a7f8d68ef

Member repositories

Repository	Role	Health v2
patrick-kidger/jaxtyping	main	90

For agents

markdown · JSON · MCP: product_card(name="patrick-kidger/jaxtyping")

Data as of 2026-10-19T08:05:51.212177+00:00 · Report a problem