mlx_stft
