Lead AI Engineer (MLX)