NumPyro Distribution

LuasNumPyro([gp, var_dict, likelihood_fn, ...])

Custom NumPyro distribution which allows a luas.GPClass.GP object to be used with NumPyro.