Make use of pylearn2.utils.grad