feat(gh-2142): add VSBC diagnostic

Author: michaelellis003

numpyro/infer/__init__.py

@@ -3,6 +3,7 @@
 
 
 from numpyro.infer.barker import BarkerMH
+from numpyro.infer.calibration import vsbc_diagnostic
 from numpyro.infer.elbo import (
     ELBO,
     RenyiELBO,
@@ -41,6 +42,7 @@
     "init_to_value",
     "log_likelihood",
     "reparam",
+    "vsbc_diagnostic",
     "BarkerMH",
     "DiscreteHMCGibbs",
     "ELBO",