netket.jax.vjp

Contents

netket.jax.vjp#

netket.jax.vjp(fun, *primals, has_aux=False, conjugate=False)[source]#

Overloads:

fun (Callable[…, T]), primals (Any), has_aux (Literal[False]), conjugate (bool) → tuple[T, Callable]
fun (Callable[…, tuple[T, U]]), primals (Any), has_aux (Literal[True]), conjugate (bool) → tuple[T, Callable, U]

Parameters:

fun (Callable)
has_aux (bool)
conjugate (bool)

Return type:

tuple[Any, Callable] | tuple[Any, Callable, Any]