Modifies existing model to do gradient allreduce, but doesn't change class so you don't need "module"
(module)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected