Conditional Monge Gap: JIT-compatible loss + training estimator#679

Open

DhruvaRajwade wants to merge 6 commits intoott-jax:mainfrom

DhruvaRajwade:pr-605-clean

Commits on Mar 18, 2026

fix(cmonge_gap): replace jnp.unique loop with segment interface for JIT compatibility
DhruvaRajwade
committed
feat: add ConditionalMongeGapEstimator for training condition-aware maps
DhruvaRajwade
committed
test: expand cmonge_gap test suite with equivalence and cost function tests
DhruvaRajwade
committed

Commits on Mar 23, 2026

Commits on Apr 1, 2026

style: fix docstring indentation and pytest mark style for ruff v0.4.10
DhruvaRajwade
committed