jeertmans/DiffeRT
[DOC] Document when `compute_paths` can be JIT-compiled
Open
#337 opened on Oct 23, 2025
documentationgood first issueneeds-triage
Repository metrics
- Stars
- (64 stars)
- PR merge metrics
- (PR metrics pending)
Description
Terms
- Checked the existing issues and discussions to see if my issue had not already been reported;
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)
- https://differt.eertmans.be/latest/reference/_autosummary/differt.scene.TriangleScene.html#differt.scene.TriangleScene.compute_paths
- https://differt.eertmans.be/latest/notebooks/performance_tips.html
Issue type
Documentation enhancement
Recommended fix or suggestions
No response