[torch.distributions] JIT support?

jit
1

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

Best
Tim

2

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.

3

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