I tested this on my M1 MacBook Pro 13" MacOS Big Sur, using python 3.8, numpy 1.19 (from Apple's TF 2.6.0 dependencies) and got 2.35s average.