LLMs on ANE with Flexible Inputs and States
- 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
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