mlx
v0.1.0
Search documentation of mlx
Settings
mlx_param_server
(mlx v0.1.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)
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)