AI-driven neural networks