Simple and light framework for building neural nets with different architecture