All pull requests

Pull requests

• Report shape and dtype changes on excessive retracing

  #468·sylvesterkaczmarek opened Sep 13, 2026·

  ·
  1

• Honor output axes in fake pmap

  #467·sylvesterkaczmarek opened Sep 10, 2026·

  ·

• improve(asserts): clarify bare Ellipsis usage in assert_shape error

  #464·erensh27 opened Aug 9, 2026·

  ·
  2

• Suppressions for pyrefly

  #459·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Suppressions for pyrefly

  #458·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Suppressions for pyrefly

  #457·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Suppressions for pyrefly

  #456·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Suppressions for pyrefly

  #454·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Suppressions for pyrefly

  #452·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Suppressions for pyrefly

  #451·copybara-service[bot] opened Jul 20, 2026Bot·

  ·

• Modernize type annotations per PEP 585

  #445·Muneerali199 opened Jun 14, 2026·

  ·
  1

• fix: clarify assert_shape error when bare Ellipsis is passed

  #439·PewterZz opened Mar 14, 2026·

  ·
  1

• Fix assert equal chexify jit 387

  #435·Sorata-kanda opened Jan 13, 2026·

  ·
  1

• Fix issue 389 vmap scalar assertions

  #434·Sikandar1310291 opened Jan 12, 2026·

  ·

• Add chex.variants pytest example

  #433·Sikandar1310291 opened Jan 10, 2026·

  ·

• Add example for using chex.variants with pytest

  #432·Sikandar1310291 opened Jan 10, 2026·

  ·

• Fix: Resolve super() error in dataclasses via virtual subclassing

  #431·Ashutosh0x opened Jan 6, 2026·

  ·
  1

• Fix: Add type hints to chex.dataclass

  #430·Ashutosh0x opened Jan 6, 2026·

  ·
  1

• Fix: Enable kw_only=True for chex.dataclass

  #429·Ashutosh0x opened Jan 6, 2026·

  ·
  1

• Fix: Exclude test files from package distribution

  #428·Ashutosh0x opened Jan 6, 2026·

  ·
  1

• Docs: Fix broken citation link in index.rst

  #427·Ashutosh0x opened Jan 6, 2026·

  ·
  1

• Feat: Add assert_not_both_not_none

  #426·Ashutosh0x opened Jan 6, 2026·

  ·
  1

• Fix: Modernize sharding detection and error messages for JAX 0.8.2 compatibility (#422)

  #423·Ashutosh0x opened Jan 4, 2026·

  ·
  1

  5

• Add assert_not_both_not_none (fixes #393)

  #421·Sorata-kanda opened Dec 21, 2025·

  ·
  2

• Make chexify docstring test robust to platform-specific tracebacks

  #420·pandaayushman opened Dec 21, 2025·

  ·
  1