Skip to content

Use array-api-compat for runtime namespace discovery - #228

Merged
jc-macdonald merged 1 commit into
mainfrom
feat/227-array-api-compat
Sep 27, 2026
Merged

jc-macdonald merged 1 commit into
mainfrom
feat/227-array-api-compat

Conversation

@jc-macdonald

Copy link
Copy Markdown
Member

Closes #227.

Summary:

  • centralize runtime namespace discovery in the public op_system.array_namespace helper backed by array-api-compat
  • route scalar, vectorized, PyTree, history-body, propensity, and axis-kernel paths through the shared helper
  • preserve scalar and vectorized autograd graphs by stacking native results and avoiding redundant conversion of existing arrays
  • add an optional torch extra plus Torch-gated flat, PyTree, and axis-kernel autograd contracts
  • keep the stricter static Array protocol unchanged pending coordinated protocol work in Flepimop2 #345, while documenting the broader runtime contract

Validation:

  • 536 passed, 5 skipped with JAX installed
  • Torch 2.14: 3 passed with UserWarning treated as error
  • provider: 69 passed, 1 skipped
  • strict mypy: core and provider passed
  • Ruff format and lint passed repository-wide
  • strict MkDocs build passed
  • core and provider wheel build/Twine checks passed
  • clean-wheel tests: core 509 passed, 14 skipped; provider 69 passed, 1 skipped

@jc-macdonald
jc-macdonald merged commit b5aec94 into main Sep 27, 2026
6 checks passed
@jc-macdonald
jc-macdonald deleted the feat/227-array-api-compat branch September 27, 2026 03:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Use array-api-compat namespace discovery and qualify PyTorch RHS evaluation

1 participant