> Just for reference Nvidia's flagship GPU(3090)'s FP32 performance is 35.5 TFLOPS.
In context of ML, nvidia’s flagship is the A100, which has 312 TFLOPS. You can also compare with a TPU device which has 180 TFLOPS (v2) or 420 TFLOPS (v3). You can use at least the TPU v2 reliably on colab for free.