Run convergence checks when using JAX samplers#7165

Merged

ricardoV94 merged 2 commits intopymc-devs:mainfrom

ricardoV94:run_jax_warnings

Feb 28, 2024

Commits on Feb 27, 2024

Always run divergence and treedepth checks in `run_convergence_checks`
ricardoV94
committed
Refactor duplicated jax sampler code and emit convergence warnings by default
ricardoV94
committed