mlx_dist_coordinator (mlx v0.2.0)

View Source

Summary

Functions

add_worker(Node)

broadcast_parameters(Parameters)

code_change(OldVsn, State, Extra)

distribute_data(Data, Strategy)

gather_gradients()

get_parameters()

get_status()

handle_call(Request, From, State)

handle_cast(Msg, State)

handle_info(Info, State)

init(Options)

list_workers()

remove_worker(Node)

start_link()

start_link(Options)

start_training(ModelConfig, DataConfig, TrainingConfig)

stop_training()

terminate(Reason, State)