flashinfer.cake\_mhc\_train.CakeMHCTrainFinishFunction
======================================================

.. currentmodule:: flashinfer.cake_mhc_train

.. autoclass:: CakeMHCTrainFinishFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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