Simple neural networks with torch