Search results

Improve assert_shape error message when Ellipsis is used as shape matcher

Status: Open.

#437 In google-deepmind/chex;

· OmAmbole009 opened on Feb 17, 2026

jax 0.8.2 incpatibility

Status: Open.

#422 In google-deepmind/chex;

· GaetanLepage opened on Dec 26, 2025

Assert for either is populated but not both

Status: Open.

#393 In google-deepmind/chex;

· alexlembcke opened on Apr 30, 2025

Combining assert_scalar_positive with jax.vmap

Status: Open.

#389 In google-deepmind/chex;

· thomaspinder opened on Apr 10, 2025

jitted, chexified chex.assert_equal fails for value assertions

Status: Open.

#387 In google-deepmind/chex;

· emergenz opened on Mar 25, 2025

Example of chex.variants with pytest

Status: Open.

#382 In google-deepmind/chex;

· chMoussa opened on Mar 7, 2025

Use of deprecated type hints - PEP 585

Status: Open.

#361 In google-deepmind/chex;

· jeertmans opened on Sep 19, 2024

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

Status: Open.

#355 In google-deepmind/chex;

· maming109 opened on May 19, 2024

Test files are in binary package distribution

Status: Open.

#347 In google-deepmind/chex;

· daskol opened on Apr 17, 2024

Citation link broken

Status: Open.

#329 In google-deepmind/chex;

· felixchalumeau opened on Jan 23, 2024

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

Status: Open.

#321 In google-deepmind/chex;

· carlosgmartin opened on Nov 24, 2023

Allow for nested chex.chexify

Status: Open.

#306 In google-deepmind/chex;

· nicow-elia opened on Sep 13, 2023