Image Segmentation
MLX
English
native-mlx
apple-silicon
medical-imaging
mri
brain-tumor
segmentation
nnunet
research
Instructions to use MedMLX/nnunet-brats-mlx with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use MedMLX/nnunet-brats-mlx with MLX:
# Download the model from the Hub pip install huggingface_hub[hf_xet] hf download MedMLX/nnunet-brats-mlx --local-dir nnunet-brats-mlx
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
- Atomic Chat
Download reference-comparison.json from MedMLX/nnunet-brats-mlx: direct link, hf CLI and curl.
- Browser
- Download file 8.24 kB
-
https://huggingface.co/MedMLX/nnunet-brats-mlx/resolve/main/reference-comparison.json
- Command line
-
hf download hf://MedMLX/nnunet-brats-mlx/reference-comparison.json
-
curl -L -o reference-comparison.json https://huggingface.co/MedMLX/nnunet-brats-mlx/resolve/main/reference-comparison.json
8.24 kB
| { | |
| "schema_version": "medmlx.reference.v1", | |
| "model": "nnunet-brats", | |
| "date": "2026-10-12", | |
| "case": { | |
| "dataset": "MSD Task01 BrainTumour (training)", | |
| "case_id": "BRATS_001", | |
| "input_sha256": [ | |
| "740b0adfa8de83fcbdfa5cd5f14689f6d947120bfbf057fe82954c07477bb173", | |
| "96abcdbb541c4af37f03bef466eee5d11a3f91b9a1ea10b1a4a22cb38b7b3978", | |
| "8e2de5df0abd8b25ac1136c6570a6d03f2255107ef1265145b606f07acd1f1af", | |
| "7571148ce691dccff5a84109091f43385b7c069a99cd7a967b2002aaeec64cdd" | |
| ], | |
| "inputs": [ | |
| { | |
| "role": "t1", | |
| "path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/T1w.nii.gz", | |
| "sha256": "740b0adfa8de83fcbdfa5cd5f14689f6d947120bfbf057fe82954c07477bb173", | |
| "dataset": "MSD Task01 BrainTumour (training)", | |
| "case_id": "BRATS_001", | |
| "derivation": "channel split of imagesTr/BRATS_001.nii.gz", | |
| "resolved_path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/T1w.nii.gz", | |
| "verified_sha256": "740b0adfa8de83fcbdfa5cd5f14689f6d947120bfbf057fe82954c07477bb173" | |
| }, | |
| { | |
| "role": "t1ce", | |
| "path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/T1gd.nii.gz", | |
| "sha256": "96abcdbb541c4af37f03bef466eee5d11a3f91b9a1ea10b1a4a22cb38b7b3978", | |
| "dataset": "MSD Task01 BrainTumour (training)", | |
| "case_id": "BRATS_001", | |
| "derivation": "channel split of imagesTr/BRATS_001.nii.gz", | |
| "resolved_path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/T1gd.nii.gz", | |
| "verified_sha256": "96abcdbb541c4af37f03bef466eee5d11a3f91b9a1ea10b1a4a22cb38b7b3978" | |
| }, | |
| { | |
| "role": "t2", | |
| "path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/T2w.nii.gz", | |
| "sha256": "8e2de5df0abd8b25ac1136c6570a6d03f2255107ef1265145b606f07acd1f1af", | |
| "dataset": "MSD Task01 BrainTumour (training)", | |
| "case_id": "BRATS_001", | |
| "derivation": "channel split of imagesTr/BRATS_001.nii.gz", | |
| "resolved_path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/T2w.nii.gz", | |
| "verified_sha256": "8e2de5df0abd8b25ac1136c6570a6d03f2255107ef1265145b606f07acd1f1af" | |
| }, | |
| { | |
| "role": "flair", | |
| "path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/FLAIR.nii.gz", | |
| "sha256": "7571148ce691dccff5a84109091f43385b7c069a99cd7a967b2002aaeec64cdd", | |
| "dataset": "MSD Task01 BrainTumour (training)", | |
| "case_id": "BRATS_001", | |
| "derivation": "channel split of imagesTr/BRATS_001.nii.gz", | |
| "resolved_path": "<home>/Documents/Codex/2026-10-08/https-drive-google-com-drive-folders/outputs/medmlx/datasets/qualification-2026-10-08/derived/msd-brain-channels/BRATS_001/FLAIR.nii.gz", | |
| "verified_sha256": "7571148ce691dccff5a84109091f43385b7c069a99cd7a967b2002aaeec64cdd" | |
| } | |
| ] | |
| }, | |
| "original": { | |
| "code": "nnunetv2", | |
| "version_or_commit": "2.8.1", | |
| "upstream_commit": "468cf803df9b267150ae2b6c0c59b8ac84f16227", | |
| "precision": "FP32 network; upstream FP16 Gaussian sliding-window accumulation; CPU autocast OFF", | |
| "command": "<paper-repo>/work/reference/nnunet-brats/.venv/bin/python benchmarks/reference/nnunet-brats.py --threads 2", | |
| "folds": [ | |
| 0 | |
| ], | |
| "configuration": "3d_fullres", | |
| "trainer": "nnUNetTrainer", | |
| "use_mirroring": false, | |
| "mirroring_axes": [], | |
| "tile_step_size": 0.5, | |
| "device": "cpu", | |
| "checkpoint_path": "<repositories>/research/radnn/work/nnunet-next-20261001/brats-source/fold_0/checkpoint_final.pth", | |
| "checkpoint_sha256": "82261f2482f9502249aa6eb3f05b0412a47f070afdbe91a3673c7ac5c6f55c01", | |
| "packages": { | |
| "nnunetv2": "2.8.1", | |
| "dynamic-network-architectures": "0.4.4", | |
| "torch": "2.13.0", | |
| "numpy": "2.5.2", | |
| "scipy": "1.18.1", | |
| "scikit-image": "0.26.0", | |
| "SimpleITK": "2.5.6", | |
| "nibabel": "5.4.2" | |
| }, | |
| "inference_wall_s": 101.54423004199634, | |
| "output_path": "work/reference/nnunet-brats/out/mask.nii.gz", | |
| "output_sha256": "f2d1dc2d745da71b7ae73208bedd307892e23c635dd697f90205fecc9c3b3b87" | |
| }, | |
| "mlx": { | |
| "run_record": "results/runs/nnunet-brats.json", | |
| "run_record_sha256": "2793a5a52c127df7112f99fcee831bcb610a5514bfdb4c3e3152895c76aa0eb5", | |
| "output_path": "<paper-repo>/work/runs/nnunet-brats/out/mask.nii.gz", | |
| "output_sha256": "f2d1dc2d745da71b7ae73208bedd307892e23c635dd697f90205fecc9c3b3b87", | |
| "use_mirroring": false, | |
| "mirroring_axes": [] | |
| }, | |
| "comparison": { | |
| "metric": "minimum_per_label_dice", | |
| "value": 1.0, | |
| "per_label": { | |
| "0": { | |
| "dice": 1.0, | |
| "original_voxels": 8802715, | |
| "mlx_voxels": 8802715 | |
| }, | |
| "1": { | |
| "dice": 1.0, | |
| "original_voxels": 25520, | |
| "mlx_voxels": 25520 | |
| }, | |
| "2": { | |
| "dice": 1.0, | |
| "original_voxels": 66292, | |
| "mlx_voxels": 66292 | |
| }, | |
| "4": { | |
| "dice": 1.0, | |
| "original_voxels": 33473, | |
| "mlx_voxels": 33473 | |
| } | |
| }, | |
| "threshold": 0.99, | |
| "result": "pass", | |
| "differing_voxel_count": 0, | |
| "labels_compared": "Every label present in either output, including background", | |
| "geometry": { | |
| "original": { | |
| "shape": [ | |
| 240, | |
| 240, | |
| 155 | |
| ], | |
| "affine": [ | |
| [ | |
| 1.0, | |
| 0.0, | |
| -0.0, | |
| -0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 1.0, | |
| -0.0, | |
| -0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 0.0, | |
| 1.0, | |
| 0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 1.0 | |
| ] | |
| ], | |
| "spacing": [ | |
| 1.0, | |
| 1.0, | |
| 1.0 | |
| ], | |
| "orientation": [ | |
| "R", | |
| "A", | |
| "S" | |
| ], | |
| "qform_code": 1, | |
| "sform_code": 1 | |
| }, | |
| "mlx": { | |
| "shape": [ | |
| 240, | |
| 240, | |
| 155 | |
| ], | |
| "affine": [ | |
| [ | |
| 1.0, | |
| 0.0, | |
| -0.0, | |
| -0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 1.0, | |
| -0.0, | |
| -0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 0.0, | |
| 1.0, | |
| 0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 1.0 | |
| ] | |
| ], | |
| "spacing": [ | |
| 1.0, | |
| 1.0, | |
| 1.0 | |
| ], | |
| "orientation": [ | |
| "R", | |
| "A", | |
| "S" | |
| ], | |
| "qform_code": 1, | |
| "sform_code": 1 | |
| }, | |
| "input": { | |
| "shape": [ | |
| 240, | |
| 240, | |
| 155 | |
| ], | |
| "affine": [ | |
| [ | |
| 1.0, | |
| 0.0, | |
| 0.0, | |
| 0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 1.0, | |
| 0.0, | |
| 0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 0.0, | |
| 1.0, | |
| 0.0 | |
| ], | |
| [ | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 1.0 | |
| ] | |
| ], | |
| "spacing": [ | |
| 1.0, | |
| 1.0, | |
| 1.0 | |
| ], | |
| "orientation": [ | |
| "R", | |
| "A", | |
| "S" | |
| ], | |
| "qform_code": 1, | |
| "sform_code": 1 | |
| } | |
| } | |
| }, | |
| "notes": "One recorded case; reproducibility comparison, not accuracy or clinical qualification. Relocated RadNN paths are recorded and verified against the original run hashes.", | |
| "failure": null | |
| } | |