I tried the model on colab and while it was slow, it wasn’t as slow as you are reporting. If you’re getting the same error across 3 machines on fresh environments, there must be a problem in the script that is going unnoticed. Again, I’d check that all the shapes are as you expect, especially the F object.