Add check_utils.py in _lapsrc. It provide a function create_check_function to test the difference between LapJAX and jax.hessian/grad for any functions. The detailed usage is described in the corresponding function doc. Also provide a test file test/test_function.py. It can test if the behavior of all wrapped functions are the same as expectation.
Add check_utils.py in _lapsrc. It provide a function create_check_function to test the difference between LapJAX and jax.hessian/grad for any functions. The detailed usage is described in the corresponding function doc. Also provide a test file test/test_function.py. It can test if the behavior of all wrapped functions are the same as expectation.