\begin{tabular}{llll}
\toprule
Model & MCTM & HMAF (B) & HMAF (S) \\
\midrule
JIT Compile & -- & True & True \\
First Joint Bijector & ScaleMatvecLinearOperator & -- & -- \\
Invert First Joint Bijector & True & -- & -- \\
Joint Bijector Conditional Event Shape & 1 & -- & -- \\
Extrapolation for Joint Bijector & True & -- & -- \\
Parameter Shape 0 for Joint Bijector & 3 & -- & -- \\
Polynomial Order for Joint Bijector & 6 & -- & -- \\
Joint Initial Learning Rate & 0.001 & 0.001 & 0.001 \\
Type of Joint Flow & -- & masked\_autoregressive\_flow\_first\_dim\_masked & masked\_autoregressive\_flow\_first\_dim\_masked \\
Number of Flows & -- & 1 & 1 \\
Joint Bijector & -- & BernsteinPolynomial & RationalQuadraticSpline \\
Joint Bijector Range Min & -- & -- & -5 \\
Interval Width for Joint Bijector & -- & -- & 10 \\
Minimum Slope for Joint Bijector & -- & -- & 0.001 \\
Minimum Bin Width for Joint Bijector & -- & -- & 0.001 \\
Number of Bins for Joint Bijector & -- & -- & 16 \\
Number of Parameters for Joint Bijectors & -- & 32 & 47 \\
Activation Function for MAF & -- & relu & relu \\
First Hidden Units for MAF & -- & 512 & 512 \\
Second Hidden Units for MAF & -- & 512 & 512 \\
Activation Function for X0 Parameters & -- & relu & relu \\
Batch Normalization for X0 Parameters & -- & False & False \\
Dropout Rate for X0 Parameters & -- & False & False \\
First Hidden Units for X0 Network & -- & 512 & 512 \\
Second Hidden Units for X0 Network & -- & 512 & 512 \\
Nested Bernstein Bijector Domain Min & 0 & -- & -- \\
Nested Bernstein Bijector Domain Max & 35 & -- & -- \\
Joint Bijector Extrapolation & -- & False & -- \\
Nested Bernstein Bijector Domain Min & -- & -5 & -- \\
Nested Bernstein Bijector Domain Max & -- & 5 & -- \\
Invert Joint Bijector & -- & True & -- \\
Joint Bijector Codomain Min & -- & -5 & -- \\
Joint Bijector Codomain Max & -- & 5 & -- \\
Third Hidden Units for MAF & -- & 512 & 512 \\
Is Conditional for MAF & -- & True & True \\
MAF Conditional Event Shape & -- & 1 & 1 \\
Third Hidden Units for X0 Network & -- & 512 & 512 \\
Is Conditional for X0 Parameters & -- & True & True \\
X0 Parameters Conditional Event Shape & -- & 1 & 1 \\
\bottomrule
\end{tabular}
