jeertmans/DiffeRT

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

开放

#337 创建于 2025年10月23日

 (0 条评论) (0 个反应) (0 位负责人)Python (10 个派生)auto 404
documentationgood first issueneeds-triage

仓库指标

星标
 (64 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

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

贡献者指南