einops
mlx
numpy
transformers