# 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)