jeertmans/DiffeRT

[DOC] Document when `compute_paths` can be JIT-compiled

Open

#337 opened on Oct 23, 2025

 (0 comments) (0 reactions) (0 assignees)Python (10 forks)auto 404
documentationgood first issueneeds-triage

Repository metrics

Stars
 (64 stars)
PR merge metrics
 (PR metrics pending)

Description

Terms

Describe the issue

As discussed in https://github.com/jeertmans/DiffeRT/pull/326#issuecomment-3437022985, the compute_paths method exhibits some challenges when it comes to JIT-compilation, as some cases, e.g., when setting disconnect_inactive_triangles to True, will create output arrays whose shapes depend on the values of the input arrays. This is not compatible with JAX's tracing system, this should probably be documented somewhere.

Affected page(s)

Issue type

Documentation enhancement

Recommended fix or suggestions

No response

Contributor guide