flashinfer.cake\_mhc\_train.CakeMHCTrainPrepareFunction
=======================================================

.. currentmodule:: flashinfer.cake_mhc_train

.. autoclass:: CakeMHCTrainPrepareFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~CakeMHCTrainPrepareFunction.boxed_grads_call
      ~CakeMHCTrainPrepareFunction.clear_saved_tensors_on_access
      ~CakeMHCTrainPrepareFunction.dirty_tensors
      ~CakeMHCTrainPrepareFunction.generate_vmap_rule
      ~CakeMHCTrainPrepareFunction.materialize_grads
      ~CakeMHCTrainPrepareFunction.metadata
      ~CakeMHCTrainPrepareFunction.needs_input_grad
      ~CakeMHCTrainPrepareFunction.next_functions
      ~CakeMHCTrainPrepareFunction.non_differentiable
      ~CakeMHCTrainPrepareFunction.output_grad_dtypes
      ~CakeMHCTrainPrepareFunction.requires_grad
      ~CakeMHCTrainPrepareFunction.saved_for_forward
      ~CakeMHCTrainPrepareFunction.saved_tensors
      ~CakeMHCTrainPrepareFunction.saved_variables
      ~CakeMHCTrainPrepareFunction.to_save