Issues

• ULP assertions raise a dtype-view error on scalar bfloat16 values

  google-deepmind/chex#475 · Filter by author sylvesterkaczmarek opened on Oct 6, 2026

• Matching NaNs fail close assertions only after JIT compilation

  google-deepmind/chex#472 · Filter by author sylvesterkaczmarek opened on Oct 3, 2026

• Improve assert_shape error message when Ellipsis is used as shape matcher

  google-deepmind/chex#437 · Filter by author OmAmbole009 opened on Feb 17, 2026
  1

• jax 0.8.2 incpatibility

  google-deepmind/chex#422 · Filter by author GaetanLepage opened on Dec 26, 2025
  1

• Assert for either is populated but not both

  google-deepmind/chex#393 · Filter by author alexlembcke opened on Apr 30, 2025

• Combining assert_scalar_positive with jax.vmap

  google-deepmind/chex#389 · Filter by author thomaspinder opened on Apr 10, 2025
  1

• jitted, chexified chex.assert_equal fails for value assertions

  google-deepmind/chex#387 · Filter by author emergenz opened on Mar 25, 2025

• Example of chex.variants with pytest

  google-deepmind/chex#382 · Filter by author chMoussa opened on Mar 7, 2025
  1

• Use of deprecated type hints - PEP 585

  google-deepmind/chex#361 · Filter by author jeertmans opened on Sep 19, 2024
  1

• ImportError: cannot import name 'DeviceLocalLayout' from 'jax._src.layout' (/opt/conda/lib/python3.10/site-packages/jax/_src/layout.py)

  google-deepmind/chex#355 · Filter by author maming109 opened on May 19, 2024
  2

• Test files are in binary package distribution

  google-deepmind/chex#347 · Filter by author daskol opened on Apr 17, 2024
  1

• Citation link broken

  google-deepmind/chex#329 · Filter by author felixchalumeau opened on Jan 23, 2024
  1

• chex.dataclass wrapper causes type error: Expected no arguments to dataclass constructor

  google-deepmind/chex#321 · Filter by author carlosgmartin opened on Nov 24, 2023
  1

• Allow for nested chex.chexify

  google-deepmind/chex#306 · Filter by author nicow-elia opened on Sep 13, 2023
  1

• TypeError: non-default argument 'value' follows default argument

  google-deepmind/chex#305 · Filter by author George-Ogden opened on Aug 31, 2023
  1

• Numpy Conflict

  google-deepmind/chex#298 · Filter by author nickknyc opened on Aug 1, 2023

• Mypy index type error with chex.dataclass

  google-deepmind/chex#296 · Filter by author hylkedonker opened on Jul 20, 2023

• Analog to flax.struct.dataclass

  google-deepmind/chex#248 · Filter by author homerjed opened on Mar 3, 2023
  2

• chex.Dimensions API enhancement

  google-deepmind/chex#231 · Filter by author wbrenton opened on Jan 27, 2023
  1

• post_init error in inherited dataclass

  google-deepmind/chex#212 · Filter by author thomaspinder opened on Nov 16, 2022
  1

• Dill pickling chex.dataclass blows the stack

  google-deepmind/chex#188 · Filter by author JonyEpsilon opened on Sep 30, 2022

• Better error report for max traces exceeded

  google-deepmind/chex#176 · Filter by author uduse opened on Jul 16, 2022
  2

• Typing issue with chex.dataclass

  google-deepmind/chex#155 · Filter by author wookayin opened on Apr 15, 2022
  2

• fake_pmap_and_jit has a confusing interface

  google-deepmind/chex#144 · Filter by author marcvanzee opened on Feb 11, 2022

• chex.variants(with_pmap=True) ignores static_argnames

  google-deepmind/chex#115 · Filter by author fvisin opened on Nov 23, 2021
  9