Support zero gradients functionality #52
Description
Now gradients are overwritten for each call, need to accumulate instead and provide an API to zero them.
{{ message }}
Now gradients are overwritten for each call, need to accumulate instead and provide an API to zero them.