* Final step of backward, return a map from param_name to grad

* Complete the final step of backward

Return the param_name to grad_info