Co-authored-by: Xin Yao <xiny@nvidia.com>
state_dict
load_state_dict
param_groups
dgl.optim.SparseGradOptimizer