Module @jax-js/jax

Namespaces

lax
nn
numpy
random
scipySpecial
tree

Enumerations

DType

Classes

Array
ClosedJaxpr
Jaxpr

Interfaces

JsTreeDef

Type Aliases

Device
JsTree
OwnedFunction

Variables

devices

Functions

blockUntilReady
defaultDevice
devicePut
grad
hessian
init
jacfwd
jacrev
jit
jvp
linearize
makeJaxpr
setDebug
valueAndGrad
vjp
vmap

References

jacobian → jacrev