docs: def grad_named(fun, argname):
      '''Takes gradients with respect to a named argument.
         Doesn't work on *args or **kwargs.'''
      arg_index = getargspec(fun).args.index(argname)
      return grad(fun, arg_index)