Is the AMX accelerator used on Apple silicon?

[philipturner]

I forgot to hyperlink the following comment on this thread. The AMX needs to use all 8 power cores for full utilization.

How to check mps availability?

Is AMX already used in pytorch ? I saw you posted a question asking that and also this comment on GitHub. I would regard whoever posted that comment with skepticism. Regardless, if PyTorch uses Accelerate, then it uses the AMX. @albanD said that PyTorch should use Accelerate by default, although the documentation does not officially confirm this is true. For example, it seems to say that other math libraries will be chosen instead of Accelerate if they already exist on the system, because the…