Create diff_test_jit.py