Replace some torch.vmap usage with a hand-vectorized BatchedNMCLogPredictiveLikelihood
#1287
Run time
Learn about OS # on GitHub Actions| Job | Run time |
|---|---|
| 3m 7s | |
| 3m 14s | |
| 4m 21s | |
| 10m 42s |