PHASE Turbulence MR
Multi-regime PHASE model for 2-D incompressible MHD decaying turbulence, with gated Re/Rm adapters and per-regime diffusion normalization.
- Channels: [ux, uy, Bx, By]
- Reynolds numbers: Re=Rm in {80, 200, 400, 650, 1000, 1500, 2050, 2750, 3600, 4500}
- Conditioner:
conditioner/model.safetensors - Diffusion model:
diffusion/model.safetensors - Normalization:
normalization/
POSEIDON dependency
git clone https://github.com/camlab-ethz/poseidon.git external/poseidon
git -C external/poseidon checkout b8fa28f59bd7f7673323f28d11a12c6f3a215c61
pip install --no-deps -e external/poseidon
Load
import torch
from phase import PHASEPipeline
model = PHASEPipeline.from_pretrained(
"phaseMHD/PHASE-Turbulence-MR",
device="cuda" if torch.cuda.is_available() else "cpu",
)
# Physical initial condition: [batch, channels, x, y]
# The turbulence models were trained on 26 frames from t=0 to t=1.
times = torch.linspace(0.0, 1.0, 26, device=model.device)
prediction = model.predict(initial_fields, times, re=1000, seed=0)
# prediction: [batch, channels, time, x, y]
Install the runtime from PHASE:
pip install "phase[hub,dino,scot] "
The package contains model-only weights, exact inference configs, and training-derived normalization statistics. See pipeline.yaml and provenance.json for the artifact mapping.