Nicolas Emmenegger

icon company ETHZ icon location Zürich CH icon location Interested in sequential decision making, optimization, and statistics

Results 1 issues of Nicolas Emmenegger

Make Distribution pytree work under jax.vmap

8 comment

I have a function which returns a posterior distribution of type `Independent` with `base_dist` being `Normal`. I want to `vmap` over the function in order to get the posteriors for...

help wanted