[P2] Mac chip MPS mode support

[frankaging]

Descriptions:
Currently, matrix exponential will fail with M1 chip MPS framework for rotation-based interventions. Need to figure out other ways to handle this so that M1 chip can run pyvene with trainable interventions like DAS as well.

[fblissjr]

even better, MLX support? :)