Senior AI Engineer - Remote

Overview

We are looking for an experienced AI/ML Engineer with deep expertise in feature engineering and model engineering to join our team supporting a global email production suite used across Google Workspace (Gmail). The core mission is to build and optimize machine learning models that can detect and prevent threats embedded in email content, attachments, and metadata.

In Short

• Design and implement robust feature engineering pipelines to extract meaningful signals from structured and unstructured email data.
• Build and fine-tune deep learning and traditional ML models to detect malicious intent in emails (e.g., phishing, malware).
• Collaborate with security analysts and data engineers to define threat patterns and data labeling strategies.
• Use MLFlow to manage the complete machine learning lifecycle including experimentation, reproducibility, deployment, and monitoring.
• Optimize Spark-based data processing pipelines to efficiently handle large-scale datasets.
• Conduct offline and online evaluations of model performance using appropriate metrics.
• Stay updated on the latest advancements in AI/ML and apply them to continuously enhance the detection models.

Requirements

• Proven expertise in feature engineering and model development for NLP, anomaly detection, or security domains.
• Strong programming skills in Python with hands-on experience in PyTorch for model development.
• Experience with Apache Spark (PySpark) for distributed data processing.
• Deep understanding of MLFlow for model tracking, versioning, and deployment.
• Strong background in machine learning theory, model evaluation, and productionization.
• Experience working with large datasets, especially from enterprise platforms like Google Workspace.
• Knowledge of common threat detection techniques and email-based attack vectors is a plus.

Benefits

• Work on a high-impact platform used by millions of Gmail users globally.
• Collaborate with a talented team of engineers, data scientists, and security professionals.
• Opportunity to shape next-generation email threat detection using cutting-edge AI/ML.