maxent_grpo.training.runtime.deepspeed

DeepSpeed and Accelerate integration helpers.

Functions

_maybe_create_deepspeed_plugin()

Construct a DeepSpeedPlugin from Accelerate env/config when available.

get_trl_prepare_deepspeed()

Return TRL's prepare_deepspeed helper when available.

require_deepspeed(context[, module])

Return a DeepSpeed module import or raise a contextual RuntimeError.

maxent_grpo.training.runtime.deepspeed.get_trl_prepare_deepspeed()[source]

Return TRL’s prepare_deepspeed helper when available.

Return type:

Any | None

maxent_grpo.training.runtime.deepspeed.require_deepspeed(context, module='deepspeed')[source]

Return a DeepSpeed module import or raise a contextual RuntimeError.

Parameters:
Return type:

Any