AI model architecture