A Neural Network modules written in Swift