amici.AMICIModule
=================

.. currentmodule:: amici

.. autoclass:: AMICIModule
   :members:
   :inherited-members:
   :show-inheritance:

   
   
   .. rubric:: Methods

   .. autosummary::
      :nosignatures:
   
      ~AMICIModule.__init__
      ~AMICIModule.add_caching_hooks
      ~AMICIModule.add_hook
      ~AMICIModule.add_module
      ~AMICIModule.add_perma_hook
      ~AMICIModule.apply
      ~AMICIModule.bfloat16
      ~AMICIModule.buffers
      ~AMICIModule.cache_all
      ~AMICIModule.cache_some
      ~AMICIModule.check_and_add_hook
      ~AMICIModule.check_hooks_to_add
      ~AMICIModule.children
      ~AMICIModule.clear_contexts
      ~AMICIModule.compile
      ~AMICIModule.cpu
      ~AMICIModule.cuda
      ~AMICIModule.double
      ~AMICIModule.eval
      ~AMICIModule.extra_repr
      ~AMICIModule.float
      ~AMICIModule.forward
      ~AMICIModule.generative
      ~AMICIModule.get_buffer
      ~AMICIModule.get_caching_hooks
      ~AMICIModule.get_extra_state
      ~AMICIModule.get_parameter
      ~AMICIModule.get_submodule
      ~AMICIModule.half
      ~AMICIModule.hook_points
      ~AMICIModule.hooks
      ~AMICIModule.inference
      ~AMICIModule.ipu
      ~AMICIModule.load_state_dict
      ~AMICIModule.loss
      ~AMICIModule.modules
      ~AMICIModule.mtia
      ~AMICIModule.named_buffers
      ~AMICIModule.named_children
      ~AMICIModule.named_modules
      ~AMICIModule.named_parameters
      ~AMICIModule.on_load
      ~AMICIModule.parameters
      ~AMICIModule.register_backward_hook
      ~AMICIModule.register_buffer
      ~AMICIModule.register_forward_hook
      ~AMICIModule.register_forward_pre_hook
      ~AMICIModule.register_full_backward_hook
      ~AMICIModule.register_full_backward_pre_hook
      ~AMICIModule.register_load_state_dict_post_hook
      ~AMICIModule.register_load_state_dict_pre_hook
      ~AMICIModule.register_module
      ~AMICIModule.register_parameter
      ~AMICIModule.register_state_dict_post_hook
      ~AMICIModule.register_state_dict_pre_hook
      ~AMICIModule.remove_all_hook_fns
      ~AMICIModule.requires_grad_
      ~AMICIModule.reset_hooks
      ~AMICIModule.run_with_cache
      ~AMICIModule.run_with_hooks
      ~AMICIModule.sample
      ~AMICIModule.set_extra_state
      ~AMICIModule.set_submodule
      ~AMICIModule.setup
      ~AMICIModule.share_memory
      ~AMICIModule.state_dict
      ~AMICIModule.to
      ~AMICIModule.to_empty
      ~AMICIModule.train
      ~AMICIModule.type
      ~AMICIModule.xpu
      ~AMICIModule.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~AMICIModule.T_destination
      ~AMICIModule.call_super_init
      ~AMICIModule.device
      ~AMICIModule.dump_patches
      ~AMICIModule.training