mlx_dist_worker (mlx v0.1.0)

View Source

Summary

Functions

code_change(OldVsn, State, Extra)

compute_gradients()

get_gradients()

get_status()

handle_call(Request, From, State)

handle_cast(Msg, State)

handle_info(Info, State)

init(Options)

initialize_training(ModelConfig, TrainingConfig)

register_with_coordinator(CoordinatorNode)

set_data(Data, BatchIndex)

start_link()

start_link(Options)

stop_training()

terminate(Reason, State)

update_parameters(Parameters)