[错误] Jit 相关的梯度错误

`JitTrace_ELBO` 在使用 `MultivariateNormal` 分布时，由 `scale_tril` 参数化，会产生与梯度相关的错误。我尚未测试其他输入组合，但下面提供了一个 MWE，可以在我的系统和 Google Colab 上重现此错误（下面引用了操作系统和包版本）。如果使用 `Trace_ELBO`，则同一个 MWE不会产生此错误。