jactorch.graph

Submodules

jactorch.graph.context	Context manager used in a forward pass.
jactorch.graph.nn_env
jactorch.graph.parameter	Utilities to access, filter, and mark parameters in a torch.nn.Module.
jactorch.graph.variable