QA Engineer – AI/ML

Key Responsibilities

• Develop and maintain automated test suites for ML pipelines and data workflows using Python and Pytest
• Validate data integrity and model performance using SQL and statistical metrics
• Design QA strategies for scalable ML model deployment (batch, real-time, and streaming)
• Ensure reproducibility, traceability, and auditability of ML experiments using MLflow
• Collaborate with data scientists and ML engineers to identify testing bottlenecks early in the development cycle
• Build and maintain continuous testing pipelines within AWS (especially with SageMaker, Lambda, S3, etc.)
• Conduct performance testing of APIs and model endpoints in production
• Help enforce best practices for version control, test coverage, and CI/CD integration

Required Qualifications

• 3–6 years of QA or SDET experience in software or ML environments
• Strong programming skills in Python (unit testing, scripting, API testing)
• Proficiency in SQL for data validation, transformations, and test scenario building
• Experience with Pytest or other test frameworks
• Solid understanding of machine learning lifecycle, model validation, and data pipelines
• Experience with MLflow for experiment tracking and model management
• Hands-on experience with AWS services, especially SageMaker, S3, Lambda, EC2
• Familiar with CI/CD pipelines and DevOps tools (e.g., GitHub Actions, Jenkins)

Job Types: Full-time, Permanent

Pay: Up to ₹2,000,000.00 per year

Benefits:

• Health insurance
• Provident Fund

Application Question(s):

• Where do you stay in Bangalore?
• What is your current Notice Period?

Work Location: In person