flashinfer.cake\_mhc\_det\_train.CakeMHCDetTrainHeadFunction
============================================================

.. currentmodule:: flashinfer.cake_mhc_det_train

.. autoclass:: CakeMHCDetTrainHeadFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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