AI neural network implementation