add neural network into AIModule