ml-explore / ml-explore/mlx-examples
Can I convert nnU-Net pytorch model to MLX?
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 9k
- Forks
- 1.2k
- PR merge metrics
- No merged PRs in 30d
Description
Hi all!
Is it possible to convert a medical image segmentation model trained using the nnU-Net framework and stored as a .pth file into an MLX compatible format?
Thanks!
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
The issue mentions an nnU-Net PyTorch model stored as a .pth file, but names no repository file, test, or entry point. First clarify whether conversion is supported and what MLX-compatible output and validation criteria would define completion.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100