patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
PythonNOASSERTION
Issues
- 2
- 9
Compatibility with multiprocessing / joblib - AttributeError: type object 'Float[Tensor, 'batch_size num_classes']' has no attribute 'index_variadic'
#198 opened by jaanli - 2
Broken link
#245 opened by lockwo - 5
Batched pytree dataclasses
#242 opened by JeyRunner - 5
- 2
feature request - benchmark using https://github.com/RightTyper/RightTyper - would be helpful @onefact !
#244 opened by jaanli - 2
Functions without type hints and import hook
#197 opened by nimashoghi - 1
`jaxtyped` Annotation fails
#187 opened by dxm447 - 1
v0.2.26 Release Failed
#183 opened by asford - 2
Inconsistent shape checking for lists of tensors
#234 opened by seanroelofs - 7
Is it possible for method annotation of an eqx.Module to pick up instance field annotations?
#241 opened by jjyyxx - 5
Static typing issues with ndarray
#243 opened by srush - 3
- 4
AbstractDType Example
#239 opened by srush - 1
Is it possible to use `AbstractDtype` to support syntax of not `Float[Array, "dim1 dim2"]` but instead `FloatArr["dim1 dim2"]`?
#238 opened by stu-twelvelabs - 4
issue with torch.compile
#235 opened by f-fuchs - 4
- 2
Leading integer and ellipses in pytree raises error
#227 opened by smorad - 2
Annotations for tensors with dynamics dimensions
#225 opened by martenlienen - 3
Installing `jaxtyping` makes Pytest slow(er)
#220 opened by thejcannon - 1
add type annotation for jaxtyped
#217 opened by nahaharo - 1
`install_import_hook` skip `no_type_check`
#215 opened by nstarman - 1
numpy structured dtype support
#204 opened by alexfanqi - 6
- 8
Can typeguard be an optional dependency?
#199 opened by z4m0 - 4
Unions not working
#210 opened by kvablack - 4
Incompatibility with flax.linen.tabulate
#209 opened by kvablack - 2
Bug with default argument binding
#206 opened by jaraujo98 - 3
- 5
Issues with torch.compile
#196 opened by botev - 1
How to use with Sphinx autodoc?
#195 opened by benkoppe - 1
Move equinox "tree_pformat" into jaxtyping or allow users to configure their own
#194 opened by vasiliykarasev - 4
Question: manual assertion
#192 opened by JuanFMontesinos - 4
jax dependency error when jax is not installed
#188 opened by jsternabsci - 3
Random instances / Hypothesis-like generation
#189 opened by srush - 3
Support for NestedTensors
#185 opened by jaanli - 1
Allow two variadic shapes when it makes sense
#184 opened by kho - 4
BUG: Exception if array in args is modified in a function with the same name
#172 opened by JuanFMontesinos - 2
Symbolic expressions example doesn't run
#181 opened by kvablack - 4
einops-like packing notation
#180 opened by MilesCranmer - 2
Disabling JAX import
#178 opened by MilesCranmer - 3
- 2
Isinstance checks against a ShapeDtypeStruct
#173 opened by tttc3 - 3
Multi-backend tensor support in jaxtyping for frameworks like Keras 3 and einops
#168 opened by anas-rz - 7
Support typechecking of jax.sharding.NamedSharding
#164 opened by reinerp - 1
- 3
- 1
- 3
Can shape information be preserved by the LSP ?
#163 opened by Doublonmousse - 2
Undefined name in Ruff: F821
#162 opened by krzysztofrusek