slapo.framework_dialect.deepspeed.engine

Functions:

init_ds_engine(model, **kwargs)

Initialize the DeepSpeed engine.

slapo.framework_dialect.deepspeed.engine.init_ds_engine(model, **kwargs)[source]

Initialize the DeepSpeed engine.