4 ms·Mostly from the tight JAX-TPU integration yeahby MasterScrat 2y agoMostly from the tight JAX-TPU integration yeah