About this role
Come work at a place where innovation and teamwork come together to support the most exciting missions in the world!
Job Description
We are seeking a Data Science focused QA engineer to develop next-generation Security Analytics products. You will work closely with Data scientists, engineers and product managers to design and optimize AI driven security solutions.
As QA engineer, the ideal candidate has a strong background in Backend engineering, system integrations, ML,AI and data pipelines.
Responsibilities (QA Engineer – Data Science / ML)
• Establish QA best practices for Traditional ML and Generative AI workflows, including:
• Functional and regression testing of ML pipelines using pytest and Airflow/Dagster test utilities and API testing tools (e.g., Postman, pytest-httpx).
• Validate data contracts, schemas, and API compatibility across services using Pandera, and custom validation rules.
• Model behavior validation (input/output ranges, invariants, edge cases) using NumPy, SciPy, and statistical assertions
• Runtime and performance testing for inference latency, throughput, and resource usage using Locust, k6, or custom load tests.
• Integrate ML-specific tests into CI/CD pipelines using GitHub Actions, GitLab CI, or Jenkins, alongside containerized workflows (Docker, Kubernetes).
• Implement LLM-specific testing, including:
• Prompt and response validation, determinism checks, and regression testing using LangSmith.
• Evaluation of hallucinations, toxicity, and policy adherence using LLM-as-a-judge and/or rule-based checks.
• Cost, token usage, and timeout monitoring for GenAI workflows
• Verify logging, monitoring, and alerting for ML services using Prometheus, Grafana, and cloud-native observability tools.
Requirements:
• BS or MS in Computer Science or a related field.
• 2-5 years of experience in Data or Machine Learning projects.
• Familiarity and experience of GenAI applications and tools -PyTorch, LangChain, vLLM etc.
• Demonstrates a commitment to continuous learning in this rapidly evolving field.
• Tools listed in the responsibilities section.