Try #4

abhijeetgangan · May 22, 2024 · 6d2c3d0 · 6d2c3d0
1 parent b7b21d0
commit 6d2c3d0
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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