TPU-powered simulation