Comment by pedropaulovc

Where is this perf gain coming from? Running on TPUs?