Mixing DiscreteHMCGibbs and HMCGibbs [NumPyro]

numpyro
1

Hi,

According to this, we have class

  • HMCGibbs combines HMC/NUTS steps with custom Gibbs updates. Gibbs updates must be specified by the user.

  • DiscreteHMCGibbs combines HMC/NUTS steps with Gibbs updates for discrete latent variables. The corresponding Gibbs updates are computed automatically.

But is it possible to have both functions within one numpyro model?

Thanks,
Edwin