All pull requests

Pull requests

• Preserve strict dtype equality under chexify

  #479·Linxiushen opened Oct 20, 2026·

  ·

• Clarify fake_pmap_and_jit flags and warn on positional use

  #478·sylvesterkaczmarek opened Oct 11, 2026·

  ·
  1

• Reject unsupported named static arguments for pmap test variants

  #477·sylvesterkaczmarek opened Oct 10, 2026·

  ·
  1

• Handle scalar bfloat16 leaves in ULP assertions

  #476·sylvesterkaczmarek opened Oct 6, 2026·

  ·
  1

• Match eager NaN handling in compiled tree close assertions

  #473·sylvesterkaczmarek opened Oct 3, 2026·

  ·
  1

• Reject NaN in scalar sign assertions

  #471·dhairyajangir opened Oct 2, 2026·

  ·
  1

• Fix tuple reconstruction of non-mappable dataclasses

  #469·Linxiushen opened Sep 27, 2026·

  ·
  1

• Report shape and dtype changes on excessive retracing

  #468·sylvesterkaczmarek opened Sep 13, 2026·

  ·
  1

  2

• 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·5 tasks done·

  ·

• 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