afnio.optim.tgd

afnio.optim.tgd(params, grads, momentum_buffer_list, model_client, messages, inputs, constraints, momentum, **completion_args)

Functional API that performs TGD (Textual Gradient Descent) algorithm computation.

See SGD for details.