File size: 304 Bytes
ab4488b |
1 2 3 4 5 6 |
# Exposes API for extending autograd
from .tracer import Box, primitive, register_notrace, notrace_primitive
from .core import (SparseObject, VSpace, vspace, VJPNode, JVPNode,
defvjp_argnums, defvjp_argnum, defvjp,
defjvp_argnums, defjvp_argnum, defjvp, def_linear)
|