Version dependant slowing down of Gaussian Mixture sampling in Ubuntu 20.04

I’ve updated the snippet, it should use @register_canonicalize instead of @register_specialize

Locally, that fixes the speed differences for me

1 Like