[torch.distributions] JIT support?

does anyone know whether there is some information somewhere about whether/when torch.Distributions will receive jit support?


As far as I know, you can use torch.jit.trace on torch.Distributions. And there are tests covering this use case. torch.jit.script, however, doesn’t fully work with torch.Distributions yet. Hope this helps.

I can’t directly answer that, but from viewing sources, I would say a complete redesign/rewrite would be needed for TorchScript conformance, unfortunately.