Solve tokamak MHD equilibrium for a D-shaped plasma

from apex import Aether

simos = Aether(api_key=os.environ["APEX_API_KEY"])

# Solve tokamak MHD equilibrium for an ITER-like D-shaped plasma, with
# couplings to the divertor thermal model, breeding-blanket neutronics, and
# magnet structural model. Reproduces the methodology from the August 2025
# release.
eq = simos.simulate(
    domain="mhd",
    geometry="iter-like-d-shape",
    plasma={
        "Bt_T": 5.3,
        "Ip_MA": 15.0,
        "q95": 3.0,
        "ne_avg_m3": 1.0e20,
        "Te_keV": 8.0,
    },
    couplings=[
        "divertor_thermal",
        "blanket_neutronics",
        "magnet_structural",
    ],
    convergence={"max_outer_iters": 40, "tol": 1e-5},
)

print(f"q_axis = {eq.q_axis:.3f}")
print(f"q_95   = {eq.q_95:.3f}")
print(f"peak divertor flux = {eq.peak_divertor_flux_mw_m2:.2f} MW/m^2")
print(f"blanket tritium breeding ratio = {eq.tbr:.3f}")
print(f"max magnet stress = {eq.max_magnet_stress_MPa:.0f} MPa")