在DiffTaichi中如何忽略一些自动的导数展开项

大家好,我有一个关于 difftaichi 的问题。我希望在 diff 模拟中忽略一些导数展开项。有人有示例或教程吗?

动机:一些模拟方法存在自动微分爆炸的问题,例如 SPH。我希望在 taichi 中实现 DiffFR [Li et al., ToG23]。