"import mlx as torch"