[Optional] - AOS as differentiable nn.Module

[SchmidtPhilipp]

Pytorch implements autograd. Implement the AOS preprocessor using pytorch functions as an nn.Module which allows to learn the focal distances which are initialized as weights of this layer.
The framework pytorch3d implements a LFR.