mlx
v0.2.0
Search documentation of mlx
Settings
mlx_dist_worker
(mlx v0.2.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)
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)