Update mlx_config.py