PyTorch implementation of inference aware neural optimisation (de Castro and Dorigo, 2018 https://www.sciencedirect.com/science/article/pii/S0010465519301948)
