core.nn.nsf.create_nsf_wrapped(**kwargs)
Wraps the NSF model in a FlowWrapper. This is required for parallel training, and wraps the log_prob method as a forward method.