Replace some torch.vmap usage with a hand-vectorized BatchedNMCLogPredictiveLikelihood
#1288
Run time
Learn about OS # on GitHub Actions| Job | Run time |
|---|---|
| 14m 52s | |
| 17m 49s | |
| 15m 1s | |
| 47m 42s |