functional

jacfwd(callable: Callable) -> Callable

jacfwd(func: fn(List[Array]) raises -> List[Array]) -> Callable

jacrev(callable: Callable, remat: Bool = False) -> Callable

jacrev(func: fn(List[Array]) raises -> List[Array], remat: Bool = False) -> Callable

grad(callable: Callable, remat: Bool = False) -> Callable

grad(func: fn(List[Array]) raises -> List[Array], remat: Bool = False) -> Callable

jit(func: Callable) -> Callable

jit(func: fn(List[Array]) raises -> List[Array]) -> Callable

jvp(func: Callable, primals: List[Array], tangents: List[Array]) -> Tuple[List[Array], List[Array]]

jvp(func: fn(List[Array]) raises -> List[Array], primals: List[Array], tangents: List[Array]) -> Tuple[List[Array], List[Array]]

vjp(func: Callable, primals: List[Array], remat: Bool = False) -> Tuple[List[Array], Callable]

vjp(func: fn(List[Array]) raises -> List[Array], primals: List[Array], remat: Bool = False) -> Tuple[List[Array], Callable]

vmap(func: Callable, in_axes: List[Int] = List(), out_axes: List[Int] = List()) -> Callable

vmap(func: fn(List[Array]) raises -> List[Array], in_axes: List[Int] = List(), out_axes: List[Int] = List()) -> Callable