Senior AI Data Engineer - ML Pipelines & Governance

Altimetrik; New York (NY); Hybrid; USD 140,000 - 190,000; Full time

Job summary

Altimetrik is seeking an AI Data Engineer to design and optimize machine learning data pipelines, focusing on model tracking, lifecycle management, and governance integration. This role blends data engineering with ML expertise to support our data and AI initiatives.

You will build Databricks-based pipelines, implement feature engineering, model versioning, and real-time monitoring, collaborating with data scientists and ML engineers across our enterprise.

Qualifications

• Bachelor's or Master's degree in CS, data science, ML, or related field.
• 10-15 years in data engineering; at least 2 years on AI/ML workloads.
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow, scikit-learn).
• Strong experience with Databricks, Spark, distributed ML workflows.
• Understanding of ML lifecycle, training, deployment, monitoring.
• Knowledge of vector databases, embeddings, and similarity search.
• Proficiency in SQL and data governance / AI ethics principles.

Responsibilities

• Design and implement data pipelines for AI model metadata, training data lineage, and metrics.
• Build data infrastructure on Databricks leveraging Spark for large-scale processing.
• Develop MCP servers and enable AI data distribution via MCP.
• Develop feature engineering pipelines and data preprocessing workflows.
• Implement model versioning, experiment tracking, and model registry integration with MLflow.
• Create automated workflows for AI agent discovery, classification, and inventory management.
• Design and maintain knowledge graph structures for AI model relationships and data lineage.
• Build real-time data pipelines for model monitoring, drift detection, and performance tracking.
• Develop data quality frameworks for AI training and validation data.
• Collaborate with data scientists to optimize data access patterns and feature stores.
• Implement security and compliance controls for sensitive AI data and artifacts.
• Create documentation for AI data architectures, schemas, and integration patterns.

Skills

Python

ML Frameworks

Databricks

Apache Spark

Model governance

Data pipelines

Feature engineering

SQL

Data quality

Collaboration

AI ethics

Education

Bachelor's or Master's in CS/Data Science/ML

Tools

Databricks

MLflow

Apache Spark

Job description

Altimetrik is seeking an AI Data Engineer to design and optimize machine learning data pipelines, focusing on model tracking, lifecycle management, and governance integration. This role blends data engineering with ML expertise to support our data and AI initiatives.

You will build Databricks-based pipelines, implement feature engineering, model versioning, and real-time monitoring, collaborating with data scientists and ML engineers across our enterprise.