The full NSF: stacked spline-autoregressive transforms with order reversal

Usage

nsf_module(
  dim_x,
  dim_theta,
  n_transforms,
  hidden,
  n_bins,
  tail_bound,
  embedding = NULL
)