3 ms·Jax was built with TPUs in mind, so it's not surprising that they use TPUsby anvuong 2y agoJax was built with TPUs in mind, so it's not surprising that they use TPUs