\begin{tabular}{ll}
\toprule
 & MAF (B) \\
\midrule
JIT Compile & False \\
Epochs & 400 \\
Batch Size & 512 \\
Learning Rate Decay Steps & epochs \\
Initial Learning Rate & 0.001 \\
Learning Rate Scheduler & cosine\_decay \\
Early Stopping & 50 \\
Validation Split & 0.250 \\
Output Distribution & masked\_autoregressive\_flow \\
Number of Flows & 2 \\
Bijector & BernsteinPolynomial \\
Bernstein Bijector Extrapolation & False \\
Bernstein Bijector Domain Min & -5 \\
Bernstein Bijector Domain Max & 5 \\
Invert Bijector & True \\
Allow Flexible Bounds for Parameters & False \\
Bounds constraints for Parameters & linear \\
Nested Bernstein Bijector Codomain Min & -5 \\
Nested Bernstein Bijector Codomain Max & 5 \\
Number of Parameters & 128 \\
Activation Function & relu \\
First Hidden Units & 128 \\
Second Hidden Units & 128 \\
Third Hidden Units & 128 \\
Layer Overwrite 0 Domain Min & 0 \\
Layer Overwrite 0 Domain Max & 1 \\
\bottomrule
\end{tabular}
