apple / apple/coremltools

Pytorch LayerNorm convert to .mlmodel

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

Description

## ❓Question

Pytorch LayerNorm convert to .mlmodel, cann't run in GPU,because it has been broken down into many operators.

![Image](https://github.com/user-attachments/assets/a64d4efc-97ae-40fd-be99-c6ba7fc2d58b)

Contributor guide

Open the contributing guide

Research direction

The issue names no files, tests, or entry points. Start by reproducing the PyTorch LayerNorm conversion to an .mlmodel and checking why the converted operators cannot run on the GPU; done means the cause is identified and the conversion is validated for GPU execution.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Bug
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.