Skip to content

API Reference

utorch is organized into three primary modules:

  • utorch.autograd: Reverse-mode automatic differentiation engine, computation graph nodes, and the Tensor abstraction.
  • utorch.nn: Neural network modules (Module, Parameter), layers (Linear), loss functions, and optimizers.
  • utorch.utils: Data loaders, batching utilities, and computational graph visualization tools.