13 Commits (67013463981d6394fd90f1200fd9b84c7bfedbe2)

Author SHA1 Message Date
  Megvii Engine Team 6701346398 docs(mge): fix some docstring format problem 4 years ago
  Megvii Engine Team 0d4568d6f8 feat(mge): rename broadcast -> broadcast_to 4 years ago
  Megvii Engine Team 026af62042 docs(mge): docs typo fix 4 years ago
  Megvii Engine Team e6715910c5 docs(mge/imperative): add docstring for GradManager 4 years ago
  Megvii Engine Team 10e942d952 refactor(mge): polish api 4 years ago
  Megvii Engine Team c7acba41fc refactor(mge/optimizer): refine gradmanager api, record = __enter__ 4 years ago
  Megvii Engine Team 8c482b6709 fix(mge/grad): make register_after_backward_callback private 4 years ago
  Megvii Engine Team e9104ef157 fix(mge/parampack): fix copy stream, import cycle 4 years ago
  Megvii Engine Team e283663a02 fix(mge/imperative): update tests to new optimizer api 4 years ago
  Megvii Engine Team b5016b9d29 feat(mge/parampack): add parampack in allreduce callback 4 years ago
  Megvii Engine Team 5ae89c799b refactor(mgb/grad): place grad at param.grad 4 years ago
  Megvii Engine Team 9faa32fcb7 fix(mge/imperative): fix grad callback 4 years ago
  Megvii Engine Team 6070266766 refactor(mge/grad_manager): refactor gradmanager, add allreduce callback 4 years ago