jaxtyping has 5.9M monthly downloads and 2K GitHub stars. Version 0.3.11 supports Python 3.11+, first published in 2022.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of wadler-lindig. It is used by 564 tracked packages, including equinox, orbax-export and lineax.
GitHub topics include jax, python-typing and typing.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- wadler-lindig >=0.1.3
Used By
- equinox
- orbax-export
- lineax
- tokamax
- optimistix
- diffrax
- google-tunix
- klujax
- tpu-inference
- sax
- torchio
- transformer-lens
- gsplat
- litert-torch
- alphagenome
- interpax
- nvidia-physicsnemo
- kauldron
- nerfstudio
- gemma
- litert-torch-nightly
- eformer
- alphafold3-pytorch
- dynamax
- uni2ts
- exponax
- haliax
- pdequinox
- levanter
- trainax
- apebench
- ai-edge-torch-nightly
- dynamiqs
- gpjax
- nvalchemi-toolkit
- uniception
- praxis
- meshgpt-pytorch
- marin-haliax
- slsqp-jax
- marin-levanter
- differt
- esm2quinox
- flowmc
- unxt
- quax
- vmecpp
- paramax
- body-models
- rc-foundry