patrick-kidger/jaxtyping

View on GitHub

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Stars: 1.9k
Forks: 96
Open beginner issues: 0
Indexed issues: 74

Dominant language: Python
License: No license data
Last GitHub push: Jul 8, 2026
Latest indexed: Sep 20, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed
PR merge metrics: No merged PRs in 30d

Beginner friendly Everything

Open All

74 open issues indexed

wrapped_fn recompiles and hits torch._dynamo config.recompile_limit Open

question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/jaxtyping#384 · 2 comments · Opened Mar 16, 2026
Class-level scoping question Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#383 · 1 comment · 1 reaction · Opened Mar 4, 2026
[Feature Request] `+` to match "at least one"/"1 or more" array axes Open

Difficulty 4/5 3-5 days Newbie friendliness 42/100

patrick-kidger/jaxtyping#378 · Opened Feb 3, 2026
Is it practical to use axis names in runtime functionality? Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/jaxtyping#377 · 2 comments · Opened Feb 3, 2026
dataset.map(jaxtyped_function, fn_kwargs={"model": model}) will be very slow. Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#375 · 1 comment · Opened Jan 29, 2026
Feature Request: Allow for unbound composite structures as return types Open

question

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/jaxtyping#357 · 2 comments · Opened Nov 23, 2025
Pydantic for runtime type checking minimal example Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#354 · 1 comment · Opened Oct 22, 2025
Jaxtyping annotations don't work with pyserde Open

question

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#352 · 5 comments · Opened Oct 21, 2025
Feature Request: Support for astropy.units.Quantity types in array annotations Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#351 · 3 comments · Opened Oct 2, 2025
Cloudpickle interaction: TypeError: cannot pickle 'weakref.ReferenceType' object Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/jaxtyping#343 · 4 comments · Opened Sep 12, 2025
Support for type checking dataclass properties Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#342 · 2 comments · Opened Sep 1, 2025
TypeError: <module '__main__'> is a built-in module Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#340 · 5 comments · Opened Aug 15, 2025
Problem with symbolic shapes Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#338 · 4 comments · Opened Aug 12, 2025
Is it possible to support tuple unpacking? Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#337 · 6 comments · 2 reactions · Opened Aug 5, 2025
invalid-type-form when using ty Open

question

Difficulty 2/5 1-3 hours Newbie friendliness 35/100

patrick-kidger/jaxtyping#336 · 2 comments · 1 reaction · Opened Aug 3, 2025
How to specify per leaf shape of a pytree? Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#334 · 6 comments · Opened Jul 17, 2025
cloudpickle + weakref + index_variadic Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#332 · 4 comments · Opened Jun 26, 2025
Disable jaxtyping for individual tests Open

question

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#331 · 2 comments · Opened Jun 24, 2025
Identical jaxtyping types are not == (as of v0.2.35) Open

Difficulty 3/5 1-2 days Newbie friendliness 42/100

patrick-kidger/jaxtyping#330 · 2 comments · Opened Jun 17, 2025
import hook can poison the `pyc` cache Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#329 · 11 comments · Opened Jun 16, 2025
`Array` type doesn't deal nicely with `Tuple`s Open

question

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/jaxtyping#327 · 3 comments · Opened Jun 10, 2025
is there any way to use jaxtyping but not other type checks? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/jaxtyping#323 · 1 comment · Opened May 14, 2025
https://docs.kidger.site/jaxtyping appears to be down? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 15/100

patrick-kidger/jaxtyping#321 · 1 comment · Opened May 10, 2025
Type hinting for `eqx.Module` created under `vmap` Open

feature

Difficulty 5/5 Over a week Newbie friendliness 30/100

patrick-kidger/jaxtyping#320 · 2 comments · Opened May 6, 2025
Possible regression with dataclasses validation Open

question

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/jaxtyping#318 · 1 comment · Opened Apr 23, 2025
[Feature request] Optional dimensions Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#312 · 3 comments · Opened Mar 20, 2025
Jaxtyping a class with mutable shapes Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#303 · 1 comment · Opened Feb 28, 2025
Better type hint for `PyTree` Open

Difficulty 2/5 1-3 hours Newbie friendliness 42/100

patrick-kidger/jaxtyping#300 · 1 comment · Opened Feb 19, 2025
Annotating functions that remove a specified dimenion Open

Difficulty 5/5 Over a week Newbie friendliness 32/100

patrick-kidger/jaxtyping#298 · 1 comment · Opened Feb 17, 2025
Improve re-raised exception messages Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#279 · 1 comment · Opened Dec 23, 2024
Incompatibility with JAX-native `register_dataclass` Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#277 · 5 comments · Opened Dec 11, 2024
Can dtypes of later arguments be annotated to follow the dtypes of earlier arguments? Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#276 · 1 comment · Opened Nov 29, 2024
internal `_typeguard` not picked up by `install_import_hook` Open

question

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#274 · 1 comment · Opened Nov 21, 2024
Splicing / variadic symbolic expressions Open

feature

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/jaxtyping#265 · 1 comment · Opened Nov 12, 2024
Dimension changing Open

question

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#264 · 2 comments · Opened Nov 5, 2024
Is it possible to be used with `tensordict`? Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#263 · 3 comments · Opened Nov 5, 2024
Stateful Equinox Module: how to annotate? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 32/100

patrick-kidger/jaxtyping#253 · 4 comments · Opened Oct 3, 2024
feature request - benchmark using https://github.com/RightTyper/RightTyper - would be helpful @onefact ! Open

question

Difficulty 5/5 Over a week Newbie friendliness 15/100

patrick-kidger/jaxtyping#244 · 2 comments · Opened Aug 14, 2024
Batched pytree dataclasses Open

feature

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#242 · 11 comments · 2 reactions · Opened Aug 8, 2024
Is it possible for method annotation of an eqx.Module to pick up instance field annotations? Open

question

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/jaxtyping#241 · 7 comments · Opened Aug 6, 2024
Inconsistent shape checking for lists of tensors Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#234 · 2 comments · Opened Jul 23, 2024
Creating instances of `jaxtyped` dataclasses is slow Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/jaxtyping#232 · 5 comments · Opened Jul 17, 2024
Installing `jaxtyping` makes Pytest slow(er) Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/jaxtyping#220 · 3 comments · Opened Jun 21, 2024
How to properly escape `*` and `_` when rendering docs with Sphinx Open

question

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/jaxtyping#203 · 3 comments · Opened May 8, 2024
Are pytorch named tensors supported, like in torchtyping? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/jaxtyping#200 · 2 comments · Opened Apr 15, 2024
Issues with torch.compile Open

Difficulty 4/5 3-5 days Newbie friendliness 20/100

patrick-kidger/jaxtyping#196 · 5 comments · Opened Mar 26, 2024
Move equinox "tree_pformat" into jaxtyping or allow users to configure their own Open

feature

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#194 · 1 comment · Opened Mar 19, 2024
Random instances / Hypothesis-like generation Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/jaxtyping#189 · 3 comments · Opened Mar 7, 2024
Support for NestedTensors Open

feature question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/jaxtyping#185 · 3 comments · Opened Mar 5, 2024
Allow two variadic shapes when it makes sense Open

feature

Difficulty 5/5 Over a week Newbie friendliness 32/100

patrick-kidger/jaxtyping#184 · 1 comment · Opened Mar 5, 2024

patrick-kidger/jaxtyping

Get new issues in your inbox