search

pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
https://num.pyro.ai
Apache License 2.0
2.09k stars 227 forks source link

fix singleton plate bug #1792

Closed amifalk closed 2 months ago

amifalk commented 2 months ago

Fixes #1448