Einstein puzzle solver with neural networks