We are seeking skilled AI/ML QA Specialists with strong Databricks experience to ensure the quality, reliability, and regulatory readiness of AI/ML platforms. This role will focus on end-to-end testing of CCAR and ESG projects, covering data pipelines, feature engineering, model training, validation, deployment, and monitoring.

Key Responsibilities

Model Development Platform QA :

- Validate data ingestion, feature engineering, and training pipelines built on Databricks (Spark, Delta, MLflow).
- Design and execute QA strategies for dataset quality, schema validation, lineage, feature consistency, drift checks, and reproducibility.
- Test MLflow experiments, model versioning, and artifacts for completeness and traceability.
- Ensure compliance with model risk management (MRM), audit, and documentation standards.

Model Execution / Production Platform QA

- Test model deployment pipelines, including batch and real-time model execution.
- Validate model scoring accuracy, performance, data contracts, SLAs, error handling, and fallback logic.
- Perform regression, performance, and volume testing for production workloads.





Automation & Tooling

- Build and maintain automated test frameworks for data and ML pipelines (Databricks notebooks, PySpark, Python).
- Implement data-driven QA checks (DQ rules, nulls, thresholds, statistical validation).
- Integrate QA into CI/CD pipelines for ML workflows.

Required Skills

- 5 - 8+ years of QA or data validation experience.
- Hands-on experience with Databricks (Spark/PySpark, Delta Lake, MLflow).
- Robust Python experience for testing and automation.
- Solid understanding of the ML lifecycle.
- Knowledge of cloud platforms (Azure preferred).

Preferred Skills

- Experience with model risk management (MRM) or regulated environments.
- Exposure to feature stores, model monitoring, and drift detection.
- Experience with performance testing at scale in distributed environments.

Education

- Bachelors or Masters degree in Computer Science, Data Science, Engineering, or a related field.

(ref:hirist.tech)

📌 InfoBeans - AI/ML Quality Assurance Specialist - Python (Indore)
🏢 InfoBeans
📍 Indore