it’s pretty impressive that PyTorch is only 7% slower than this given it can be used so generally