apple / apple/coremltools

LLMs on ANE with Flexible Inputs and States

Open
#2,600 1 comment 0 reactions 0 assignees View on GitHub
question
Dominant language
Python
Stars
5.4k
Forks
850
Avg merge
4d 5h
Merged PRs (30d)
10

Description

## ❓Question
Hello,

I am trying to run the Llama-3.2-3B on the ANE on my M2 Max mac running macos 15.6.1.
Is the LLM inference with KV cache as a state, flexible input ranges, and int4 quantization/palettization supported on CPU+Neural Engine? If so, could you please point me to an example explaining the flow?

I tried to use this [tutorial](https://machinelearning.apple.com/research/core-ml-on-device-llama) (describing the deployment flow on GPU) as reference, but I could not load the converted model on ANE. All the nodes are falling back to CPU (I verified the compute unit mapping through Xcode).

Thank you.

Contributor guide

Open the contributing guide

Research direction

Start with the linked Core ML on-device Llama tutorial and reproduce the conversion on the stated M2 Max and macOS 15.6.1 environment, checking the Xcode compute-unit mapping. Determine whether KV-cache states, flexible input ranges, and int4 quantization or palettization can run on CPU plus Neural Engine; done requires a documented answer or a focused example clarifying the supported flow.

Written by the indexing model from the issue text.

Assessment

Tech stack
machine-learning, python
Domain
machine-learning, tooling
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.