AI Data Engineer

TalentOla; Princeton (NJ); Hybrid; USD 140,000 - 210,000; Full time

Job summary

TalentOla is seeking a Senior AI Data Engineer to design and optimize AI data pipelines in Princeton, NJ and NYC with a hybrid work model. You will specialize in model metadata, data lineage, and governance while integrating ML frameworks for scalable ML workflows.

The role requires 10+ years in data engineering with 2+ years in AI/ML workloads, deep Python expertise, and hands-on experience with Databricks, Spark, and MLflow for model tracking and deployment.

Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or related field.
• 5-7 years of hands-on experience in data engineering, with at least 2 years focused on AI/ML workloads.
• Expert proficiency in Python and experience with ML frameworks such as PyTorch, TensorFlow, or scikit-learn.
• Strong experience with Databricks, Apache Spark, and distributed computing for ML workflows.
• Deep understanding of the machine learning lifecycle, including model training, deployment, and monitoring processes.
• Experience with feature engineering, data preprocessing techniques, and ML data pipelines.
• Knowledge of vector databases, embeddings, and similarity search for AI applications.
• Proficiency in SQL for structured and unstructured data management.
• Understanding of data governance, model governance, and AI ethics principles.
• Strong analytical and problem-solving capabilities with attention to data quality.
• Excellent collaboration skills for working with data scientists, ML engineers, and architects.

Responsibilities

• Design and implement data pipelines for AI model metadata and data lineage.
• Build data infrastructure on Databricks leveraging Spark for large-scale processing.
• Develop MCP servers and enable AI data distribution via MCP.
• Create feature engineering pipelines and data preprocessing workflows for AI model training and inference.
• Implement model versioning, experiment tracking, and model registry integration using MLflow.
• Create automated workflows for AI agent discovery, classification, and inventory management.
• Design knowledge graph structures for AI model relationships and data lineage.
• Build real-time data pipelines for AI model monitoring and drift detection.
• Develop data quality frameworks for AI training datasets 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 and integration patterns.

Skills

Python

Databricks

Apache Spark

PyTorch

TensorFlow

scikit-learn

SQL

Data governance

Feature engineering

Data preprocessing

Vector databases

Embeddings

Knowledge graphs

MLflow

Model versioning

Education

Bachelor's or Master's in CS/DS/ML

Tools

MLflow

Neo4j

HuggingFace

Job description

Senior AI Data Engineer
Location- Princeton, NJ & NYC, NY (Hybrid)

Fulltime

Need 10+ years of experience candidates

AI Data Engineer

Role Overview

The AI Data Engineer will specialize in building and optimizing machine learning data pipelines, focusing on AI model tracking, lifecycle management, and integration with AI governance systems. This role combines data engineering expertise with AI/ML knowledge to support the organization's broader data and AI infrastructure initiatives.

Key Responsibilities

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

Required Skills and Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or related field.
• 5-7 years of hands-on experience in data engineering, with at least 2 years focused on AI/ML workloads.
• Expert proficiency in Python and experience with ML frameworks such as PyTorch, TensorFlow, or scikit-learn.
• Strong experience with Databricks, Apache Spark, and distributed computing for ML workflows.
• Deep understanding of the machine learning lifecycle, including model training, deployment, and monitoring processes.
• Experience with feature engineering, data preprocessing techniques, and ML data pipelines.
• Knowledge of vector databases, embeddings, and similarity search for AI applications.
• Proficiency in SQL for structured and unstructured data management.
• Understanding of data governance, model governance, and AI ethics principles.
• Strong analytical and problem-solving capabilities with attention to data quality.
• Excellent collaboration skills for working with data scientists, ML engineers, and architects.

Preferred/Nice-to-Have Skills

• Experience with generative AI applications, including RAG (Retrieval-Augmented Generation) and fine-tuning.
• Knowledge of LangChain, HuggingFace, or other GenAI frameworks.
• Familiarity with Azure ML, AWS SageMaker, or Google Vertex AI platforms.
• Experience with graph databases (Neo4j, Amazon Neptune) for knowledge graph implementation.
• Understanding of AI model explainability and interpretability techniques.
• Experience with A/B testing frameworks for ML model evaluation.
• Certification in Databricks, AWS, Azure, or GCP AI/ML services.
• Publications or contributions to open-source ML projects