Learning production functions for supply chains with graph neural networks