You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I wondered what the status on this is - I'm trying to do SVI with discrete latents, and I think the errors I'm getting (tracer leaks that make it impossible to run) come from this bug. I guess I have 3 questions:
Am I right to think that the fixes in #2002 deal with this problem, or is it actually #1999?
Is the fork in a state where it's reasonable to install from and use it?
And is there anything I could do to help you get the PR over the line?
Activity
fehiepsi commentedon Mar 7, 2025
The reason seems to be caused by this line
numpyro/numpyro/contrib/funsor/discrete.py
Line 59 in b766706
where the stateful adjoint tape is not compatible with jax scan.
Switching back to lazy interpretations seems to fix the leakage but it makes some tests failing.
pscicluna commentedon Jun 10, 2025
I wondered what the status on this is - I'm trying to do SVI with discrete latents, and I think the errors I'm getting (tracer leaks that make it impossible to run) come from this bug. I guess I have 3 questions:
Am I right to think that the fixes in #2002 deal with this problem, or is it actually #1999?
Is the fork in a state where it's reasonable to install from and use it?
And is there anything I could do to help you get the PR over the line?