jax

f2f0f7ef - implement commit_shared() as pass b/c of a barrier directly after it