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

.. currentmodule:: flashinfer.cake_mhc_det_train

.. autoclass:: CakeMHCDetTrainPrepareFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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