DOC: Correct a few grammatical errors

numpyro/contrib/nested_sampling.py

@@ -88,7 +88,7 @@ def _(d):
 
     def transform_fn(q):
         # NB: icdf is not available yet for Gamma distribution
-        # so this will raise an NotImplementedError for now.
+        # so this will raise a NotImplementedError for now.
         # We will need scipy.special.gammaincinv, which is not available yet in JAX
         # see issue: https://github.com/jax-ml/jax/issues/5350
         # TODO: consider wrap jaxns GammaPrior transform implementation

numpyro/distributions/truncated.py

@@ -601,7 +601,7 @@ def alpha_tangent_variable(alpha: ArrayLike) -> ArrayLike:
             alpha_tangent = jnp.where(
                 neq_neg1_mask,
                 log_x + alpha_tangent_variable(neq_neg1_alpha),
-                # Approximate derivate with right an lefthand approximation
+                # Approximate derivative with right and lefthand approximation
                 log_x
                 + (
                     alpha_tangent_variable(alpha - delta_eq_neg1)

test/contrib/test_enum_elbo.py

@@ -493,7 +493,7 @@ def test_elbo_enumerate_plate_4(outer_obs, inner_obs, scale):
     #  |       \                |
     #  | b ---> inner_obs   N=2 |
     #  +------------------------+
-    # This tests two different observations, one outside and one inside an plate.
+    # This tests two different observations, one outside and one inside a plate.
     params = {}
     params["probs_a"] = jnp.array([0.4, 0.6])
     params["probs_b"] = jnp.array([0.6, 0.4])