The PYP models benchmark for a given sample size can be summarized in the table below, with the cells being compile times (if applicable) and execution times. Timings for forward passes and gradient computations of the models will also be computed if high-level API is available and exposed. The appropriate accompanying data visualizations (line plots, surface plots) will be included to make the results more interpretable. Links will be included to MWEs.
[ ] Make a dynamic table which changes N. (e.g. N=50, 100, 200, 400, 800, 1600).
[ ] Make interactive line plots where
x-axis is N, and y-axis is execution time or compile time
be able to select which implementations / PPL to show on the plot
Model (PPL), sample size = N
Collapsed Gibbs
SMC
Particle Gibbs
HMC/NUTS (SB)
ADVI (SB)
PYP mixture of Gaussians (Turing)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of non-Gaussians (Turing)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of Gaussians (STAN)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of non-Gaussians (STAN)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of Gaussians (Nimble)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of non-Gaussians (Nimble)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of Gaussians (Pyro)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of non-Gaussians (Pyro)
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
[ ] TODO
PYP mixture of Gaussians (TFP)
NA
NA
NA
[ ] TODO
[ ] TODO
PYP mixture of non-Gaussians (TFP)
NA
NA
NA
[ ] TODO
[ ] TODO
Notes:
Prefer doing the experiments by column as opposed to by row in this table. This way, if we are run out of time, we have benchmarks across PPLs, for a few implementations; and not benchmarks across a few implementations for one or two PPLs.
For now, for fix truncation level for number of stick-breaking components. Time permitting, see if a benchmark can be done to include varying truncation levels.
The PYP models benchmark for a given sample size can be summarized in the table below, with the cells being compile times (if applicable) and execution times. Timings for forward passes and gradient computations of the models will also be computed if high-level API is available and exposed. The appropriate accompanying data visualizations (line plots, surface plots) will be included to make the results more interpretable. Links will be included to MWEs.
Notes: