feat(gh-2142): add VSBC diagnostic (#2170)

* feat(gh-2142): add VSBC diagnostic

* chore: format VSBC diagnostic files

* refactor(gh-2142): expose calibration module instead of vsbc_diagnostic

Author: michaelellis003

numpyro/infer/__init__.py

@@ -29,11 +29,12 @@
 from numpyro.infer.svi import SVI
 from numpyro.infer.util import Predictive, log_likelihood
 
-from . import autoguide, reparam
+from . import autoguide, calibration, reparam
 
 __all__ = [
     "AIES",
     "autoguide",
+    "calibration",
     "init_to_feasible",
     "init_to_mean",
     "init_to_median",