tfep.app.cartesianmaf.CartesianMAFMap
=====================================

.. currentmodule:: tfep.app.cartesianmaf

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

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~CartesianMAFMap.__init__
      ~CartesianMAFMap.add_module
      ~CartesianMAFMap.all_gather
      ~CartesianMAFMap.apply
      ~CartesianMAFMap.are_axes_atoms_mapped
      ~CartesianMAFMap.backward
      ~CartesianMAFMap.bfloat16
      ~CartesianMAFMap.buffers
      ~CartesianMAFMap.children
      ~CartesianMAFMap.clip_gradients
      ~CartesianMAFMap.compile
      ~CartesianMAFMap.configure_callbacks
      ~CartesianMAFMap.configure_flow
      ~CartesianMAFMap.configure_gradient_clipping
      ~CartesianMAFMap.configure_model
      ~CartesianMAFMap.configure_optimizers
      ~CartesianMAFMap.configure_sharded_model
      ~CartesianMAFMap.cpu
      ~CartesianMAFMap.create_dataset
      ~CartesianMAFMap.create_partial_flow
      ~CartesianMAFMap.create_universe
      ~CartesianMAFMap.cuda
      ~CartesianMAFMap.determine_atom_indices
      ~CartesianMAFMap.double
      ~CartesianMAFMap.eval
      ~CartesianMAFMap.extra_repr
      ~CartesianMAFMap.float
      ~CartesianMAFMap.forward
      ~CartesianMAFMap.freeze
      ~CartesianMAFMap.get_buffer
      ~CartesianMAFMap.get_conditioning_indices
      ~CartesianMAFMap.get_extra_state
      ~CartesianMAFMap.get_mapped_indices
      ~CartesianMAFMap.get_nonfixed_indices
      ~CartesianMAFMap.get_parameter
      ~CartesianMAFMap.get_reference_atoms_indices
      ~CartesianMAFMap.get_submodule
      ~CartesianMAFMap.half
      ~CartesianMAFMap.inverse
      ~CartesianMAFMap.ipu
      ~CartesianMAFMap.load_from_checkpoint
      ~CartesianMAFMap.load_state_dict
      ~CartesianMAFMap.log
      ~CartesianMAFMap.log_dict
      ~CartesianMAFMap.lr_scheduler_step
      ~CartesianMAFMap.lr_schedulers
      ~CartesianMAFMap.manual_backward
      ~CartesianMAFMap.modules
      ~CartesianMAFMap.mtia
      ~CartesianMAFMap.named_buffers
      ~CartesianMAFMap.named_children
      ~CartesianMAFMap.named_modules
      ~CartesianMAFMap.named_parameters
      ~CartesianMAFMap.on_after_backward
      ~CartesianMAFMap.on_after_batch_transfer
      ~CartesianMAFMap.on_before_backward
      ~CartesianMAFMap.on_before_batch_transfer
      ~CartesianMAFMap.on_before_optimizer_step
      ~CartesianMAFMap.on_before_zero_grad
      ~CartesianMAFMap.on_fit_end
      ~CartesianMAFMap.on_fit_start
      ~CartesianMAFMap.on_load_checkpoint
      ~CartesianMAFMap.on_predict_batch_end
      ~CartesianMAFMap.on_predict_batch_start
      ~CartesianMAFMap.on_predict_end
      ~CartesianMAFMap.on_predict_epoch_end
      ~CartesianMAFMap.on_predict_epoch_start
      ~CartesianMAFMap.on_predict_model_eval
      ~CartesianMAFMap.on_predict_start
      ~CartesianMAFMap.on_save_checkpoint
      ~CartesianMAFMap.on_test_batch_end
      ~CartesianMAFMap.on_test_batch_start
      ~CartesianMAFMap.on_test_end
      ~CartesianMAFMap.on_test_epoch_end
      ~CartesianMAFMap.on_test_epoch_start
      ~CartesianMAFMap.on_test_model_eval
      ~CartesianMAFMap.on_test_model_train
      ~CartesianMAFMap.on_test_start
      ~CartesianMAFMap.on_train_batch_end
      ~CartesianMAFMap.on_train_batch_start
      ~CartesianMAFMap.on_train_end
      ~CartesianMAFMap.on_train_epoch_end
      ~CartesianMAFMap.on_train_epoch_start
      ~CartesianMAFMap.on_train_start
      ~CartesianMAFMap.on_validation_batch_end
      ~CartesianMAFMap.on_validation_batch_start
      ~CartesianMAFMap.on_validation_end
      ~CartesianMAFMap.on_validation_epoch_end
      ~CartesianMAFMap.on_validation_epoch_start
      ~CartesianMAFMap.on_validation_model_eval
      ~CartesianMAFMap.on_validation_model_train
      ~CartesianMAFMap.on_validation_model_zero_grad
      ~CartesianMAFMap.on_validation_start
      ~CartesianMAFMap.optimizer_step
      ~CartesianMAFMap.optimizer_zero_grad
      ~CartesianMAFMap.optimizers
      ~CartesianMAFMap.parameters
      ~CartesianMAFMap.predict_dataloader
      ~CartesianMAFMap.predict_step
      ~CartesianMAFMap.prepare_data
      ~CartesianMAFMap.print
      ~CartesianMAFMap.register_backward_hook
      ~CartesianMAFMap.register_buffer
      ~CartesianMAFMap.register_forward_hook
      ~CartesianMAFMap.register_forward_pre_hook
      ~CartesianMAFMap.register_full_backward_hook
      ~CartesianMAFMap.register_full_backward_pre_hook
      ~CartesianMAFMap.register_load_state_dict_post_hook
      ~CartesianMAFMap.register_load_state_dict_pre_hook
      ~CartesianMAFMap.register_module
      ~CartesianMAFMap.register_parameter
      ~CartesianMAFMap.register_state_dict_post_hook
      ~CartesianMAFMap.register_state_dict_pre_hook
      ~CartesianMAFMap.requires_grad_
      ~CartesianMAFMap.save_hyperparameters
      ~CartesianMAFMap.set_extra_state
      ~CartesianMAFMap.set_submodule
      ~CartesianMAFMap.setup
      ~CartesianMAFMap.share_memory
      ~CartesianMAFMap.state_dict
      ~CartesianMAFMap.teardown
      ~CartesianMAFMap.test_dataloader
      ~CartesianMAFMap.test_step
      ~CartesianMAFMap.to
      ~CartesianMAFMap.to_empty
      ~CartesianMAFMap.to_onnx
      ~CartesianMAFMap.to_torchscript
      ~CartesianMAFMap.toggle_optimizer
      ~CartesianMAFMap.train
      ~CartesianMAFMap.train_dataloader
      ~CartesianMAFMap.training_step
      ~CartesianMAFMap.transfer_batch_to_device
      ~CartesianMAFMap.type
      ~CartesianMAFMap.unfreeze
      ~CartesianMAFMap.untoggle_optimizer
      ~CartesianMAFMap.val_dataloader
      ~CartesianMAFMap.validation_step
      ~CartesianMAFMap.xpu
      ~CartesianMAFMap.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~CartesianMAFMap.CHECKPOINT_HYPER_PARAMS_KEY
      ~CartesianMAFMap.CHECKPOINT_HYPER_PARAMS_NAME
      ~CartesianMAFMap.CHECKPOINT_HYPER_PARAMS_TYPE
      ~CartesianMAFMap.T_destination
      ~CartesianMAFMap.automatic_optimization
      ~CartesianMAFMap.call_super_init
      ~CartesianMAFMap.current_epoch
      ~CartesianMAFMap.device
      ~CartesianMAFMap.device_mesh
      ~CartesianMAFMap.dtype
      ~CartesianMAFMap.dump_patches
      ~CartesianMAFMap.example_input_array
      ~CartesianMAFMap.fabric
      ~CartesianMAFMap.global_rank
      ~CartesianMAFMap.global_step
      ~CartesianMAFMap.hparams
      ~CartesianMAFMap.hparams_initial
      ~CartesianMAFMap.local_rank
      ~CartesianMAFMap.logger
      ~CartesianMAFMap.loggers
      ~CartesianMAFMap.n_conditioning_atoms
      ~CartesianMAFMap.n_conditioning_dofs
      ~CartesianMAFMap.n_fixed_atoms
      ~CartesianMAFMap.n_mapped_atoms
      ~CartesianMAFMap.n_mapped_dofs
      ~CartesianMAFMap.n_nonfixed_atoms
      ~CartesianMAFMap.n_nonfixed_dofs
      ~CartesianMAFMap.on_gpu
      ~CartesianMAFMap.strict_loading
      ~CartesianMAFMap.trainer
      ~CartesianMAFMap.dataset
      ~CartesianMAFMap.training