einops
mlx
numpy
transformers>=4.35
torch