diff --git a/pkgs/development/python-modules/numpyro/default.nix b/pkgs/development/python-modules/numpyro/default.nix
index 89d0c2e0d18b..8decec762030 100644
--- a/pkgs/development/python-modules/numpyro/default.nix
+++ b/pkgs/development/python-modules/numpyro/default.nix
@@ -90,6 +90,30 @@ buildPythonPackage rec {
     "test_kl_dirichlet_dirichlet"
     "test_kl_univariate"
     "test_mean_var"
+    # since jax update to 0.5.0
+    "test_analytic_kl_2"
+    "test_analytic_kl_3"
+    "test_apply_kernel"
+    "test_beta_bernoulli"
+    "test_biject_to"
+    "test_bijective_transforms"
+    "test_change_point_x64"
+    "test_cholesky_update"
+    "test_dais_vae"
+    "test_discrete_gibbs_multiple_sites_chain"
+    "test_entropy_categorical"
+    "test_gaussian_model"
+    "test_get_proposal_loc_and_scale"
+    "test_guide_plate_contraction"
+    "test_kernel_forward"
+    "test_laplace_approximation_warning"
+    "test_log_prob_gradient"
+    "test_logistic_regression"
+    "test_logistic_regression_x64"
+    "test_scale"
+    "test_scan_svi"
+    "test_stein_particle_loss"
+    "test_weight_convergence"
 
     # Tests want to download data
     "data_load"