symforce.codegen.backends.pytorch package#
*THIS MODULE IS EXPERIMENTAL*
Backend for PyTorch. This generates Python functions that just call PyTorch ops, i.e. each SymForce op in your expression tree becomes a PyTorch op.
It’s possible we could do significantly better than this by generating custom PyTorch ops instead.
This currently only supports vector inputs and outputs, we do not have geo or cam types for PyTorch yet.