tfep.potentials.psi4.Psi4PotentialEnergyFunc
============================================

.. currentmodule:: tfep.potentials.psi4

.. autoclass:: Psi4PotentialEnergyFunc
   :members:
   :show-inheritance:
   :inherited-members: Module,LightningModule

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~Psi4PotentialEnergyFunc.__init__
      ~Psi4PotentialEnergyFunc.apply
      ~Psi4PotentialEnergyFunc.backward
      ~Psi4PotentialEnergyFunc.forward
      ~Psi4PotentialEnergyFunc.jvp
      ~Psi4PotentialEnergyFunc.mark_dirty
      ~Psi4PotentialEnergyFunc.mark_non_differentiable
      ~Psi4PotentialEnergyFunc.mark_shared_storage
      ~Psi4PotentialEnergyFunc.maybe_clear_saved_tensors
      ~Psi4PotentialEnergyFunc.name
      ~Psi4PotentialEnergyFunc.register_hook
      ~Psi4PotentialEnergyFunc.register_prehook
      ~Psi4PotentialEnergyFunc.save_for_backward
      ~Psi4PotentialEnergyFunc.save_for_forward
      ~Psi4PotentialEnergyFunc.set_materialize_grads
      ~Psi4PotentialEnergyFunc.setup_context
      ~Psi4PotentialEnergyFunc.vjp
      ~Psi4PotentialEnergyFunc.vmap
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~Psi4PotentialEnergyFunc.dirty_tensors
      ~Psi4PotentialEnergyFunc.generate_vmap_rule
      ~Psi4PotentialEnergyFunc.materialize_grads
      ~Psi4PotentialEnergyFunc.metadata
      ~Psi4PotentialEnergyFunc.needs_input_grad
      ~Psi4PotentialEnergyFunc.next_functions
      ~Psi4PotentialEnergyFunc.non_differentiable
      ~Psi4PotentialEnergyFunc.requires_grad
      ~Psi4PotentialEnergyFunc.saved_for_forward
      ~Psi4PotentialEnergyFunc.saved_tensors
      ~Psi4PotentialEnergyFunc.saved_variables
      ~Psi4PotentialEnergyFunc.to_save