The batched version (shape=10) can be considerably faster as it produces vectorized logp and gradient code so it’s strongly suggested over separating the terms.
It’s not just a convenience
The batched version (shape=10) can be considerably faster as it produces vectorized logp and gradient code so it’s strongly suggested over separating the terms.
It’s not just a convenience