Error: "trying to differentiate twice a function that was marked with @once_differentiable" when using LoTDEncoding independently.

[YunxuanMao]

Hi, thank you for your excellent work. I encountered this problem when using LoTDEncoding as a positional encoder for my own model when optimizing the scene representation with depth loss and photometric loss. Do you know how to solve the problem?