xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Misc bug fixes in Zero optimizer: handling differentiable argument, optimizer_dtype

Open amithrm opened this issue 1 year ago • 0 comments

This a cumulative PR with misc bug fixes and updates to Zero Redundancy Optimizer from all the authors (AWS): Guangtai Huang, Rahul Solanki, Fei Wu, Amith Mamidala

amithrm avatar Feb 02 '24 00:02 amithrm