jax

535d984f - add gpu determinism note