tfep.potentials.tblite.TBLitePotentialEnergyFunc
================================================

.. currentmodule:: tfep.potentials.tblite

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

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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