菜单
Guides edits to hmfast's differentiable JAX halo model (HaloModel, mass function, bias, concentration, profiles). Use when changing src/hmfast/halos/, power-spectrum methods, or halo-model consistency flags.
用 Codex 或 Claude 帮你安装 复制这段 Prompt，粘贴到 Codex、Claude 或其他助手里，让它检查 Skill 页面并帮你完成安装。
基于 SOC 职业分类
| name | halo-model-workflow |
| description | Guides edits to hmfast's differentiable JAX halo model (HaloModel, mass function, bias, concentration, profiles). Use when changing src/hmfast/halos/, power-spectrum methods, or halo-model consistency flags. |
src/hmfast/halos/halo_model.py for how Cosmology is wired (multiple _load_emulator calls at init) and how TophatVar / mcfit is instantiated.MassDefinition, halo_mass_function, halo_bias, subhalo_mass_function, and concentration defaults — HaloModel.update() preserves the immutable-style pattern via pytree flatten/unflatten.update() and _tree_flatten / _tree_unflatten so the model remains a valid pytree where intended.tests/test_halo_model.py (shapes, initialization, basic spectra).Run from repository root:
pytest tests/test_halo_model.py -v
Extend tests if you add public methods on HaloModel.
Reviews proposed changes to hmfast for correctness, JAX compatibility, API layering, test coverage, and hardware-cost surprises. Use when the user asks for a review, a sanity check on a diff, or before merging a feature branch.
Guides changes to hmfast Cosmology, emulator loading, and cosmological parameter sets. Use when editing cosmology.py, emulator_load.py, download.py, or adding new emulator_set keys.
Diagnoses a failed or stuck SLURM/HPC job for hmfast (OOM, timeout, NaNs, CUDA errors, missing outputs). Use when the user asks "why did my job fail" or shows a SLURM log.
Plans and launches an hmfast experiment (training, inference, or scan) on the cluster with safe resource limits. Use when the user wants to run something that costs nontrivial compute, especially on GPU.
Guides adding or modifying hmfast tracers and halo profiles. Use when editing src/hmfast/tracers/ or src/hmfast/halos/profiles/, or integrating new observables with HaloModel.
Drafts or revises a section of an astrophysics or cosmology manuscript based on hmfast results. Use when the user asks for a draft, an edit pass, or a publishability review of a section, including its figures.