PyCoreML

CoreML inference from Python — powered by Swift & ApplePy.

Direct access to Apple's CoreML framework with explicit compute unit selection (CPU, GPU, Neural Engine).

macOS only — requires Swift 6.0+ toolchain

Install

pip install -e .

Usage

import pycoreml

# Load and describe a model
info = pycoreml.load_model("/path/to/model.mlmodelc")
desc = pycoreml.model_description("/path/to/model.mlmodelc")

# Run prediction
result = pycoreml.predict("/path/to/model.mlmodelc", {"x": 42.0})

# Predict with explicit compute unit (the killer feature)
result = pycoreml.predict_with_options(
    "/path/to/model.mlmodelc",
    {"x": 42.0},
    "cpuAndNeuralEngine"  # or "cpuOnly", "cpuAndGPU", "all"
)

API

Examples

See pycoreml/examples/demo.py for a full demo.

License

BSD-3-Clause © Jagtesh Chadha — see LICENSE.