Any thoughts on how to enable the AMX co-processor? There are some references to handling the context switch here: https://github.com/apple/darwin-xnu/blob/main/osfmk/arm/arm_init.c#L506