Data management with AI