| # Exposes API for extending autograd | |
| from .core import ( | |
| JVPNode, | |
| SparseObject, | |
| VJPNode, | |
| VSpace, | |
| def_linear, | |
| defjvp, | |
| defjvp_argnum, | |
| defjvp_argnums, | |
| defvjp, | |
| defvjp_argnum, | |
| defvjp_argnums, | |
| vspace, | |
| ) | |
| from .tracer import Box, notrace_primitive, primitive, register_notrace | |