Merge pull request #11 from abhijeetgangan/Tersoff_unit_test

Try #4
abhijeetgangan · May 22, 2024 · 593e82c · 593e82c
2 parents ff433d6 + 6d2c3d0
commit 593e82c
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