INTEGRATOR_KWARG_UNITS

torch_sim.integrators.INTEGRATOR_KWARG_UNITS: Final[dict[Integrator, dict[str, IntegratorKwargUnits]]] = {Integrator.npt_crescale_isotropic: {'dt': IntegratorKwargUnits(factors=(('init', 98.22694788464062), ('step', 98.22694788464062))), 'external_pressure': IntegratorKwargUnits(factors=(('step', 6.241509125883258e-07),)), 'isothermal_compressibility': IntegratorKwargUnits(factors=(('init', 1602176.6208),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0))), 'tau': IntegratorKwargUnits(factors=(('step', 98.22694788464062),)), 'tau_p': IntegratorKwargUnits(factors=(('init', 98.22694788464062),))}, Integrator.npt_crescale_triclinic: {'dt': IntegratorKwargUnits(factors=(('init', 98.22694788464062), ('step', 98.22694788464062))), 'external_pressure': IntegratorKwargUnits(factors=(('step', 6.241509125883258e-07),)), 'isothermal_compressibility': IntegratorKwargUnits(factors=(('init', 1602176.6208),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0))), 'tau': IntegratorKwargUnits(factors=(('step', 98.22694788464062),)), 'tau_p': IntegratorKwargUnits(factors=(('init', 98.22694788464062),))}, Integrator.npt_langevin_anisotropic: {'alpha': IntegratorKwargUnits(factors=(('init', 0.010180505671156725),)), 'b_tau': IntegratorKwargUnits(factors=(('init', 98.22694788464062),)), 'cell_alpha': IntegratorKwargUnits(factors=(('init', 0.010180505671156725),)), 'dt': IntegratorKwargUnits(factors=(('init', 98.22694788464062), ('step', 98.22694788464062))), 'external_pressure': IntegratorKwargUnits(factors=(('step', 6.241509125883258e-07),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0)))}, Integrator.npt_langevin_isotropic: {'alpha': IntegratorKwargUnits(factors=(('init', 0.010180505671156725),)), 'b_tau': IntegratorKwargUnits(factors=(('init', 98.22694788464062),)), 'cell_alpha': IntegratorKwargUnits(factors=(('init', 0.010180505671156725),)), 'dt': IntegratorKwargUnits(factors=(('init', 98.22694788464062), ('step', 98.22694788464062))), 'external_pressure': IntegratorKwargUnits(factors=(('step', 6.241509125883258e-07),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0)))}, Integrator.npt_nose_hoover_isotropic: {'b_tau': IntegratorKwargUnits(factors=(('init', 98.22694788464062),)), 'dt': IntegratorKwargUnits(factors=(('init', 98.22694788464062), ('step', 98.22694788464062))), 'external_pressure': IntegratorKwargUnits(factors=(('step', 6.241509125883258e-07),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0))), 't_tau': IntegratorKwargUnits(factors=(('init', 98.22694788464062),))}, Integrator.nve: {'dt': IntegratorKwargUnits(factors=(('step', 98.22694788464062),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0),))}, Integrator.nvt_langevin: {'dt': IntegratorKwargUnits(factors=(('step', 98.22694788464062),)), 'gamma': IntegratorKwargUnits(factors=(('step', 0.010180505671156725),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0)))}, Integrator.nvt_nose_hoover: {'dt': IntegratorKwargUnits(factors=(('init', 98.22694788464062), ('step', 98.22694788464062))), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0))), 'tau': IntegratorKwargUnits(factors=(('init', 98.22694788464062),))}, Integrator.nvt_vrescale: {'dt': IntegratorKwargUnits(factors=(('step', 98.22694788464062),)), 'kT': IntegratorKwargUnits(factors=(('init', 1.0), ('step', 1.0))), 'tau': IntegratorKwargUnits(factors=(('step', 98.22694788464062),))}}

Unit-carrying kwargs of each integrator, derived at import time from the TimeArg/InverseTimeArg/PressureArg/InversePressureArg annotations on the registered init/step functions. torch_sim.runners.integrate() reads this to unit-convert kwargs.