AI Integration with Mechanistic Epidemiological Models