flashinfer.cake\_rmsnorm\_train.CakeRMSNormFunction
===================================================

.. currentmodule:: flashinfer.cake_rmsnorm_train

.. autoclass:: CakeRMSNormFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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