Replace some torch.vmap usage with a hand-vectorized BatchedNMCLogPredictiveLikelihood
#1289
Run time
Learn about OS # on GitHub Actions| Job | Run time |
|---|---|
| 12m 31s | |
| 13m 57s | |
| 14m 44s | |
| 41m 12s |