maxent\_grpo.training.zero\_utils
=================================

.. automodule:: maxent_grpo.training.zero_utils

   
   .. rubric:: Functions

   .. autosummary::
   
      _call_gather_fn
      _deepspeed_engine_cls
      _disable_hf_deepspeed_zero3_init
      _embedding_weight_needing_gather
      _embedding_weights_needing_gather
      _ensure_cuda_fallback
      _ensure_deepspeed_ready
      _gather_callable
      _is_deepspeed_engine
      _maybe_patch_zero_no_sync
      _maybe_zero_gather_embedding
      _maybe_zero_gather_params
      _reserve_zero_gather_params
      _zero_param_list
      _zero_param_ready_without_gather
      _zero_partitioning_gradients
      _zero_stage
      _zero_status_name
   
   .. rubric:: Classes

   .. autosummary::
   
      GatherCallable