jax

077f2b66 - Update Pallas debugging doc with TPU interpret mode + dynamic race detector.