Developer APIs#

These developer APIs extend Ray Train with custom trainers, backends, and callbacks.

Trainer base class#

DataParallelTrainer

Base class for distributed data parallel training on Ray.

Backend base classes#

Backend

Singleton for distributed communication backend.

BackendConfig

Parent class for configurations of training backend.

Trainer callbacks#

UserCallback

Callback interface for custom user-defined callbacks to handling events during training.