Initial implementation, all fp32 Tested against torch.optim.sgd
Attach a file by drag & drop or click to upload
