tfep.potentials.gromacs.GROMACSPotentialEnergyFunc ================================================== .. currentmodule:: tfep.potentials.gromacs .. autoclass:: GROMACSPotentialEnergyFunc :members: :show-inheritance: :inherited-members: Module,LightningModule .. automethod:: __init__ .. rubric:: Methods .. autosummary:: ~GROMACSPotentialEnergyFunc.__init__ ~GROMACSPotentialEnergyFunc.apply ~GROMACSPotentialEnergyFunc.backward ~GROMACSPotentialEnergyFunc.forward ~GROMACSPotentialEnergyFunc.jvp ~GROMACSPotentialEnergyFunc.mark_dirty ~GROMACSPotentialEnergyFunc.mark_non_differentiable ~GROMACSPotentialEnergyFunc.mark_shared_storage ~GROMACSPotentialEnergyFunc.maybe_clear_saved_tensors ~GROMACSPotentialEnergyFunc.name ~GROMACSPotentialEnergyFunc.register_hook ~GROMACSPotentialEnergyFunc.register_prehook ~GROMACSPotentialEnergyFunc.save_for_backward ~GROMACSPotentialEnergyFunc.save_for_forward ~GROMACSPotentialEnergyFunc.set_materialize_grads ~GROMACSPotentialEnergyFunc.setup_context ~GROMACSPotentialEnergyFunc.vjp ~GROMACSPotentialEnergyFunc.vmap .. rubric:: Attributes .. autosummary:: ~GROMACSPotentialEnergyFunc.dirty_tensors ~GROMACSPotentialEnergyFunc.generate_vmap_rule ~GROMACSPotentialEnergyFunc.materialize_grads ~GROMACSPotentialEnergyFunc.metadata ~GROMACSPotentialEnergyFunc.needs_input_grad ~GROMACSPotentialEnergyFunc.next_functions ~GROMACSPotentialEnergyFunc.non_differentiable ~GROMACSPotentialEnergyFunc.requires_grad ~GROMACSPotentialEnergyFunc.saved_for_forward ~GROMACSPotentialEnergyFunc.saved_tensors ~GROMACSPotentialEnergyFunc.saved_variables ~GROMACSPotentialEnergyFunc.to_save