how to save the values of gradient to the file?

I appreciate the hard work of all the developers. I am running the example `mnist_dnn.py` and I have read the tutorial of Objax, and found some guides for saving and loading the parameters of the neural network model.
But I want to ask how to save the values of the gradient in every batch. What should I do? 