Is callbacksupported for numpyro NUTS sampler?

[AlfredoJF]

I want to add an early stopping callback but seems numpyro sampler is not supported only pymc

Does anybody if callback in the sample method is supported for numpyro? Any example?

Thanks!

[ricardoV94]

Not that I know, but that would be a numpyro question not pymc. You can ask them in https://forum.pyro.ai/c/numpyro/9