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()

Return TRL’s prepare_deepspeed helper when available.

Return type:

Any | None

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

Return a DeepSpeed module import or raise a contextual RuntimeError.

Parameters:

• context (str)

• module (str)

Return type:

Any