message: consistency in function name..
diff: @@ -43,7 +43,7 @@ def custom_gradient(fx, gx, x, axis=(),
  h(x) = x * stop_gradient(g(x)) + stop_gradient(f(x) - x * g(x))
  ```
  - is such that `h(x) = stop(f(x))` and `grad[h(x), x] = stop_gradient(g(x)).`
  + is such that `h(x) = stop_gradient(f(x))` and `grad[h(x), x] = stop_gradient(g(x)).`
  In addition to scalar-domain/scalar-range functions, this function also
  supports tensor-domain/scalar-range functions. However, in the latter case it
  