patrick-kidger/jaxtyping

Issues

• Are pytorch named tensors supported, like in torchtyping?

  #200 opened 5 months ago by zxti
  2

• Compatibility with multiprocessing / joblib - AttributeError: type object 'Float[Tensor, 'batch_size num_classes']' has no attribute 'index_variadic'

  #198 opened a month ago by jaanli
  9

• Broken link

  #245 opened 4 days ago by lockwo
  2

• Batched pytree dataclasses

  #242 opened a month ago by JeyRunner
  5

• Creating instances of `jaxtyped` dataclasses is slow

  #232 opened 2 months ago by padix-key
  5

• feature request - benchmark using https://github.com/RightTyper/RightTyper - would be helpful @onefact !

  #244 opened 20 days ago by jaanli
  2

• Functions without type hints and import hook

  #197 opened 15 days ago by nimashoghi
  2

• `jaxtyped` Annotation fails

  #187 opened 15 days ago by dxm447
  1

• v0.2.26 Release Failed

  #183 opened 15 days ago by asford
  1

• Inconsistent shape checking for lists of tensors

  #234 opened a month ago by seanroelofs
  2

• Is it possible for method annotation of an eqx.Module to pick up instance field annotations?

  #241 opened a month ago by jjyyxx
  7

• Static typing issues with ndarray

  #243 opened 25 days ago by srush
  5

• `mypy` gives error for `np.ndarray` as container type

  #230 opened 2 months ago by padix-key
  3

• AbstractDType Example

  #239 opened a month ago by srush
  4

• Is it possible to use `AbstractDtype` to support syntax of not `Float[Array, "dim1 dim2"]` but instead `FloatArr["dim1 dim2"]`?

  #238 opened a month ago by stu-twelvelabs
  1

• issue with torch.compile

  #235 opened a month ago by f-fuchs
  4

• Misleading exception when runtime type checker is used directly

  #231 opened a month ago by padix-key
  4

• Leading integer and ellipses in pytree raises error

  #227 opened 2 months ago by smorad
  2

• Annotations for tensors with dynamics dimensions

  #225 opened 2 months ago by martenlienen
  2

• Installing `jaxtyping` makes Pytest slow(er)

  #220 opened 2 months ago by thejcannon
  3

• add type annotation for jaxtyped

  #217 opened 3 months ago by nahaharo
  1

• `install_import_hook` skip `no_type_check`

  #215 opened 3 months ago by nstarman
  1

• numpy structured dtype support

  #204 opened 3 months ago by alexfanqi
  1

• jaxtyping with JAX severely slowing down training speed

  #213 opened 3 months ago by kvablack
  6

• Can typeguard be an optional dependency?

  #199 opened 5 months ago by z4m0
  8

• Unions not working

  #210 opened 3 months ago by kvablack
  4

• Incompatibility with flax.linen.tabulate

  #209 opened 3 months ago by kvablack
  4

• Bug with default argument binding

  #206 opened 4 months ago by jaraujo98
  2

• How to properly escape `*` and `_` when rendering docs with Sphinx

  #203 opened 4 months ago by jeertmans
  3

• Issues with torch.compile

  #196 opened 5 months ago by botev
  5

• How to use with Sphinx autodoc?

  #195 opened 5 months ago by benkoppe
  1

• Move equinox "tree_pformat" into jaxtyping or allow users to configure their own

  #194 opened 5 months ago by vasiliykarasev
  1

• Question: manual assertion

  #192 opened 6 months ago by JuanFMontesinos
  4

• jax dependency error when jax is not installed

  #188 opened 6 months ago by jsternabsci
  4

• Random instances / Hypothesis-like generation

  #189 opened 6 months ago by srush
  3

• Support for NestedTensors

  #185 opened 6 months ago by jaanli
  3

• Allow two variadic shapes when it makes sense

  #184 opened 6 months ago by kho
  1

• BUG: Exception if array in args is modified in a function with the same name

  #172 opened 6 months ago by JuanFMontesinos
  4

• Symbolic expressions example doesn't run

  #181 opened 6 months ago by kvablack
  2

• einops-like packing notation

  #180 opened 6 months ago by MilesCranmer
  4

• Disabling JAX import

  #178 opened 7 months ago by MilesCranmer
  2

• bug: can't type flax.struct.dataclass with vmapped functions

  #177 opened 7 months ago by Egiob
  3

• Isinstance checks against a ShapeDtypeStruct

  #173 opened 7 months ago by tttc3
  2

• Multi-backend tensor support in jaxtyping for frameworks like Keras 3 and einops

  #168 opened 7 months ago by anas-rz
  3

• Support typechecking of jax.sharding.NamedSharding

  #164 opened 7 months ago by reinerp
  7

• Provide a way to get error diagnostics out of isinstance checks

  #167 opened 7 months ago by reinerp
  1

• Feature request: support for scipy sparse matrix shape type checking

  #166 opened 7 months ago by jaanli
  3

• How to type a "DTypeLike" argument and runtime check it ?

  #165 opened 7 months ago by pchanial
  1

• Can shape information be preserved by the LSP ?

  #163 opened 7 months ago by Doublonmousse
  3

• Undefined name in Ruff: F821

  #162 opened 7 months ago by krzysztofrusek
  2