quantax.nn.filter_vjp

quantax.nn.filter_vjp(fun: Callable, *primals, has_aux: bool = False, **vjpkwargs) → Tuple[Any, Callable] | Tuple[Any, Callable, Any]

Like equinox.filter_vjp.

The leaves in NoGradLayer are not considered as differentiable parameters.