Add JAX array support to pted and pted_coverage_test#14

Merged

ConnorStoneAstro merged 7 commits intomainfrom

copilot/make-pted-compatible-with-jax

Mar 3, 2026

Commits on Mar 3, 2026

Optimize _jax_cdist: squared-norm identity for L2, vmap for general p

Copilot
and
ConnorStoneAstro
committed
Update pted() signatures/docstrings for jax.Array; add jax optional extra in pyproject.toml

Copilot
and
ConnorStoneAstro
committed
Add cross-backend consistency tests for numpy/torch/JAX energy distance utils

Copilot
and
ConnorStoneAstro
committed