integrate

torch_sim.runners.integrate(system, model, *, integrator, n_steps, temperature, timestep, trajectory_reporter=None, autobatcher=False, **integrator_kwargs)[source]

Simulate a system using a model and integrator.

Parameters:

  • system (StateLike) – Input system to simulate

  • model (ModelInterface) – Neural network model module

  • integrator (Callable) – Integration algorithm function

  • n_steps (int) – Number of integration steps

  • temperature (float | ArrayLike) – Temperature or array of temperatures for each step

  • timestep (float) – Integration time step

  • integrator_kwargs – Additional keyword arguments for integrator

  • trajectory_reporter (TrajectoryReporter | dict | None) – Optional reporter for tracking trajectory. If a dict, will be passed to the TrajectoryReporter constructor.

  • autobatcher (BinningAutoBatcher | bool) – Optional autobatcher to use

  • **integrator_kwargs – Additional keyword arguments for integrator init function

Returns:

Final state after integration

Return type:

SimState