Fix JAX inference timing to include GPU sync#992

Abhishek21g wants to merge 1 commit into

Physical-Intelligence:mainPhysical-Intelligence/openpi:mainfrom

Abhishek21g:fix/inference-timing-gpu-syncAbhishek21g/openpi:fix/inference-timing-gpu-syncCopy head branch name to clipboard