Co-authored-by: Mufei Li <mufeili1996@gmail.com>
state_dict
load_state_dict
param_groups
dgl.optim.SparseGradOptimizer