Index

A

amax() (in module grgrjax)

C

callback_func() (in module grgrjax)

G

grgrjax module

J

jax_print() (in module grgrjax)

jvp_vmap() (in module grgrjax)

M

module grgrjax

N

newton_jax() (in module grgrjax)

newton_jax_jit() (in module grgrjax)

V

val_and_jacfwd() (in module grgrjax)

val_and_jacrev() (in module grgrjax) vjp_vmap() (in module grgrjax)