google-deepmind/chex

View on GitHub

Stars: 958
Forks: 74
Open beginner issues: 0
Indexed issues: 26
Avg merge: 21h 10m
Merged PRs (30d): 1

Dominant language: Python
License: Apache-2.0
Last GitHub push: Sep 10, 2026
Latest indexed: Sep 16, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

Beginner friendly Everything

Open All

26 open issues indexed

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

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

google-deepmind/chex#437 · 1 comment · Opened Feb 17, 2026
jax 0.8.2 incpatibility Open

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

google-deepmind/chex#422 · 1 comment · Opened Dec 26, 2025
Assert for either is populated but not both Open

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

google-deepmind/chex#393 · Opened Apr 30, 2025
Combining assert_scalar_positive with jax.vmap Open

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

google-deepmind/chex#389 · 1 comment · Opened Apr 10, 2025
jitted, chexified chex.assert_equal fails for value assertions Open

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

google-deepmind/chex#387 · Opened Mar 25, 2025
Example of chex.variants with pytest Open

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

google-deepmind/chex#382 · 1 comment · Opened Mar 7, 2025
Use of deprecated type hints - PEP 585 Open

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

google-deepmind/chex#361 · Opened Sep 19, 2024
ImportError: cannot import name 'DeviceLocalLayout' from 'jax._src.layout' (/opt/conda/lib/python3.10/site-packages/jax/_src/layout.py) Open

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

google-deepmind/chex#355 · 2 comments · Opened May 19, 2024
Test files are in binary package distribution Open

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

google-deepmind/chex#347 · 1 comment · Opened Apr 17, 2024
Citation link broken Open

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

google-deepmind/chex#329 · 1 comment · Opened Jan 23, 2024
`chex.dataclass` wrapper causes type error: Expected no arguments to dataclass constructor Open

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

google-deepmind/chex#321 · 1 comment · Opened Nov 24, 2023
Allow for nested chex.chexify Open

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

google-deepmind/chex#306 · 1 comment · 1 reaction · Opened Sep 13, 2023
TypeError: non-default argument 'value' follows default argument Open

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

google-deepmind/chex#305 · 1 comment · Opened Aug 31, 2023
Numpy Conflict Open

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

google-deepmind/chex#298 · 1 reaction · Opened Aug 1, 2023
Mypy index type error with `chex.dataclass` Open

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

google-deepmind/chex#296 · Opened Jul 20, 2023
Analog to `flax.struct.dataclass` Open

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

google-deepmind/chex#248 · 2 comments · Opened Mar 3, 2023
chex.Dimensions API enhancement Open

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

google-deepmind/chex#231 · 1 comment · 1 reaction · Opened Jan 27, 2023
post_init error in inherited dataclass Open

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

google-deepmind/chex#212 · 1 comment · Opened Nov 16, 2022
Dill pickling chex.dataclass blows the stack Open

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

google-deepmind/chex#188 · Opened Sep 30, 2022
Better error report for max traces exceeded Open

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

google-deepmind/chex#176 · 2 comments · Opened Jul 16, 2022
Typing issue with chex.dataclass Open

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

google-deepmind/chex#155 · 2 comments · 6 reactions · Opened Apr 15, 2022
fake_pmap_and_jit has a confusing interface Open

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

google-deepmind/chex#144 · Opened Feb 11, 2022
chex.variants(with_pmap=True) ignores `static_argnames` Open

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

google-deepmind/chex#115 · 9 comments · Opened Nov 23, 2021
Error with Pydantic Open

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

google-deepmind/chex#71 · Opened Jun 28, 2021
Specify non-pytree node dataclass fields Open

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

google-deepmind/chex#64 · 3 comments · 7 reactions · Opened Jun 6, 2021
without_jit=True for already jitted functions Open

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

google-deepmind/chex#22 · Opened Feb 22, 2021

google-deepmind/chex

Get new issues in your inbox