pypi/mlx-sample-metal-extension