mlx-examples/llms/mlx_lm/__init__.py