einops
mlx
numpy
transformers
torch