From 6d2c3d0f642dc201feb022c82fc1c0cce58c988e Mon Sep 17 00:00:00 2001
From: abhijeetgangan <abhijeet.gangan261997@gmail.com>
Date: Wed, 22 May 2024 13:08:55 -0700
Subject: [PATCH] Try #4

---
 tests/rigid_body_test.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tests/rigid_body_test.py b/tests/rigid_body_test.py
index f8a1665..edf90c2 100644
--- a/tests/rigid_body_test.py
+++ b/tests/rigid_body_test.py
@@ -30,7 +30,6 @@
 
 
 from jax.config import config as jax_config
-jax_config.update('jax_disable_jit', True)
 
 import jax.numpy as jnp