numpyro
numpyro copied to clipboard
Use functional interface funsor.adjoint.adjoint to avoid tracer leak
Fixes #1998
The current infer_discrete tests pass so I believe the change is equivalent.