jax

1fc5d2a5 - [0.6.0-UT] adding a QA test requirements file for the run_single_gpu test script… (#446)