jax

78157722 - clean up parallel_test.py file