gitextract_7t995rdz/ ├── .autorc ├── .dockerignore ├── .flake8 ├── .gitattributes ├── .github/ │ ├── EC2_GPU_RUNNER.md │ ├── ISSUE_TEMPLATE/ │ │ ├── bug_report.md │ │ ├── documentation.md │ │ ├── feature_request.md │ │ ├── maintenance.md │ │ └── question.md │ ├── PULL_REQUEST_TEMPLATE.md │ └── workflows/ │ ├── ci.yml │ ├── guide-notebooks-ec2.yml │ ├── kwyk-reproduction-ec2.yml │ ├── publish.yml │ ├── release.yml │ └── validate-book.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .zenodo.json ├── CHANGELOG.md ├── CITATION ├── CLAUDE.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── conftest.py ├── docker/ │ ├── README.md │ ├── cpu.Dockerfile │ └── gpu.Dockerfile ├── nobrainer/ │ ├── __init__.py │ ├── _version.py │ ├── augmentation/ │ │ ├── __init__.py │ │ ├── profiles.py │ │ ├── synthseg.py │ │ └── transforms.py │ ├── cli/ │ │ ├── __init__.py │ │ ├── main.py │ │ └── tests/ │ │ ├── __init__.py │ │ └── main_test.py │ ├── dataset.py │ ├── datasets/ │ │ ├── __init__.py │ │ ├── openneuro.py │ │ └── zarr_store.py │ ├── distributed_learning/ │ │ └── dwc.py │ ├── experiment.py │ ├── gpu.py │ ├── io.py │ ├── layers/ │ │ ├── InstanceNorm.py │ │ ├── __init__.py │ │ ├── bernoulli_dropout.py │ │ ├── concrete_dropout.py │ │ ├── gaussian_dropout.py │ │ ├── maxpool4d.py │ │ ├── padding.py │ │ └── tests/ │ │ └── __init__.py │ ├── losses.py │ ├── metrics.py │ ├── models/ │ │ ├── __init__.py │ │ ├── _constants.py │ │ ├── _utils.py │ │ ├── autoencoder.py │ │ ├── bayesian/ │ │ │ ├── __init__.py │ │ │ ├── bayesian_meshnet.py │ │ │ ├── bayesian_vnet.py │ │ │ ├── kwyk_meshnet.py │ │ │ ├── layers.py │ │ │ ├── utils.py │ │ │ ├── vwn_layers.py │ │ │ └── warmstart.py │ │ ├── generative/ │ │ │ ├── __init__.py │ │ │ ├── dcgan.py │ │ │ └── progressivegan.py │ │ ├── highresnet.py │ │ ├── meshnet.py │ │ ├── segformer3d.py │ │ ├── segmentation.py │ │ ├── simsiam.py │ │ └── tests/ │ │ └── __init__.py │ ├── prediction.py │ ├── processing/ │ │ ├── __init__.py │ │ ├── base.py │ │ ├── croissant.py │ │ ├── dataset.py │ │ ├── generation.py │ │ └── segmentation.py │ ├── research/ │ │ ├── __init__.py │ │ ├── loop.py │ │ └── templates/ │ │ ├── .gitkeep │ │ ├── prepare.py │ │ └── train_bayesian_vnet.py │ ├── slurm.py │ ├── sr-tests/ │ │ ├── __init__.py │ │ ├── conftest.py │ │ ├── test_bayesian_uncertainty.py │ │ ├── test_brain_generation.py │ │ ├── test_croissant_metadata.py │ │ ├── test_dataset_builder.py │ │ ├── test_extract_patches.py │ │ ├── test_kwyk_smoke.py │ │ ├── test_raw_pytorch_api.py │ │ ├── test_segmentation_estimator.py │ │ ├── test_synthseg_brain.py │ │ ├── test_zarr_conversion.py │ │ └── test_zarr_pipeline.py │ ├── tests/ │ │ ├── __init__.py │ │ ├── contract/ │ │ │ ├── __init__.py │ │ │ └── test_cli.py │ │ ├── gpu/ │ │ │ ├── __init__.py │ │ │ ├── test_bayesian_e2e.py │ │ │ ├── test_gan_e2e.py │ │ │ ├── test_multi_gpu.py │ │ │ └── test_predict_e2e.py │ │ ├── integration/ │ │ │ ├── __init__.py │ │ │ ├── test_datalad_commit.py │ │ │ └── test_research_smoke.py │ │ └── unit/ │ │ ├── __init__.py │ │ ├── test_bayesian_layers.py │ │ ├── test_bayesian_models.py │ │ ├── test_class_weights.py │ │ ├── test_croissant.py │ │ ├── test_dataset.py │ │ ├── test_dataset_builder.py │ │ ├── test_datasets_openneuro.py │ │ ├── test_estimator_generation.py │ │ ├── test_estimator_segmentation.py │ │ ├── test_experiment.py │ │ ├── test_generative.py │ │ ├── test_gpu.py │ │ ├── test_io_weights.py │ │ ├── test_io_zarr.py │ │ ├── test_layers.py │ │ ├── test_losses.py │ │ ├── test_metrics.py │ │ ├── test_model_interface.py │ │ ├── test_model_registry.py │ │ ├── test_models_segmentation.py │ │ ├── test_prediction.py │ │ ├── test_research_commit.py │ │ ├── test_research_loop.py │ │ ├── test_segformer3d.py │ │ ├── test_slurm.py │ │ ├── test_stride_patches.py │ │ ├── test_synthseg.py │ │ ├── test_training.py │ │ ├── test_training_convergence.py │ │ ├── test_transform_pipeline.py │ │ ├── test_vwn_layers.py │ │ ├── test_zarr_dataset.py │ │ └── test_zarr_store.py │ ├── training.py │ ├── utils.py │ └── validation.py ├── pyproject.toml └── scripts/ ├── kwyk_reproduction/ │ ├── 01_assemble_dataset.py │ ├── 02_train_meshnet.py │ ├── 03_train_bayesian.py │ ├── 04_evaluate.py │ ├── 05_compare_kwyk.py │ ├── 06_block_size_sweep.py │ ├── ARCHITECTURE.md │ ├── README.md │ ├── __init__.py │ ├── build_kwyk_manifest.py │ ├── config.yaml │ ├── config_kwyk_smoke.yaml │ ├── convert_zarr_shard.py │ ├── experiments/ │ │ ├── 01_20260330_eval_deterministic/ │ │ │ ├── README.md │ │ │ ├── eval_deterministic.py │ │ │ ├── results_summary.md │ │ │ └── run.sbatch │ │ ├── 02_20260330_binary_bayesian/ │ │ │ ├── README.md │ │ │ ├── config.yaml │ │ │ ├── eval_binary.py │ │ │ ├── eval_only.sbatch │ │ │ └── run.sbatch │ │ ├── 03_20260330_warmstart_diagnostic/ │ │ │ ├── README.md │ │ │ ├── diagnose.py │ │ │ ├── results_summary.md │ │ │ └── run.sbatch │ │ ├── 04_20260330_fixed_warmstart/ │ │ │ ├── README.md │ │ │ ├── run.py │ │ │ └── run.sbatch │ │ ├── 05_20260330_kwyk_from_scratch/ │ │ │ ├── README.md │ │ │ ├── results_summary.md │ │ │ ├── run.py │ │ │ └── run.sbatch │ │ ├── 06_20260331_fullvol_augment/ │ │ │ ├── README.md │ │ │ ├── config_256.yaml │ │ │ ├── config_256_mp.yaml │ │ │ ├── config_fullvol.yaml │ │ │ ├── run_128.sbatch │ │ │ ├── run_256.sbatch │ │ │ ├── run_256_a100.sbatch │ │ │ ├── run_256_gradckpt.sbatch │ │ │ └── run_256_mp.sbatch │ │ ├── 07_20260401_ddp_128/ │ │ │ ├── config.yaml │ │ │ └── run.sbatch │ │ ├── 08_20260401_ddp_128_full/ │ │ │ ├── config.yaml │ │ │ └── run.sbatch │ │ └── task-planner.md │ ├── label_mappings/ │ │ ├── 115-class-mapping.csv │ │ ├── 50-class-mapping.csv │ │ └── 6-class-mapping.csv │ ├── run.sh │ ├── slurm_convert_zarr.sbatch │ ├── slurm_kwyk_bayesian.sbatch │ ├── slurm_kwyk_evaluate.sbatch │ ├── slurm_kwyk_smoke.sbatch │ ├── slurm_train.sbatch │ ├── slurm_zarr_array.sbatch │ ├── submit_kwyk_smoke.sh │ └── utils.py └── synthseg_evaluation/ ├── 02_train.py ├── 03_evaluate.py ├── 04_compare.py ├── README.md ├── config.yaml ├── run.sh └── slurm_train.sbatch