docs: add doc strings to the neural networks architectures

docs: add doc strings to the neural networks architectures #16