3 ms·

You should give PyTorch a try