Show a bit more work in custom_gradient docstring
It took me a minute to understand the equivalence used in the example, so I thought it would help other readers if we show a tiny bit more of the work.