mlx_param_server (mlx v0.2.0)

View Source

Summary

Functions

aggregate_gradients(GradientsList)

apply_optimizer_step(OptimizerConfig)

checkpoint(Path)

code_change(OldVsn, State, Extra)

get_parameters()

get_parameters(Key)

get_stats()

handle_call(Request, From, State)

handle_cast(Msg, State)

handle_info(Info, State)

init(Config)

initialize_parameters(ModelSpec)

restore(Path)

start_link()

start_link(Config)

terminate(Reason, State)

update_gradients(WorkerId, Gradients)