quantax.nn.filter_grad#
- quantax.nn.filter_grad(fun: Callable, *, has_aux: bool = False, **gradkwargs) Callable | Tuple[Callable, Any] #
Creates a function that computes the gradient of
fun
similar to equinox.filter_grad.The leaves in
NoGradLayer
are not considered as differentiable parameters.