flashinfer.cake\_mhc\_train.CakeMHCTrainHeadFunction
====================================================

.. currentmodule:: flashinfer.cake_mhc_train

.. autoclass:: CakeMHCTrainHeadFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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