torch_sim.optimizers.fireΒΆ

FIRE (Fast Inertial Relaxation Engine) optimizer implementation.

Functions

fire_init

Initialize a FIRE optimization state.

fire_step

Perform one FIRE optimization step.

Classes

TYPE_CHECKING

bool(x) -> bool