Performance: MLX best practice