AI Data Engineer

$160k - $170k

Reuben Cooley, Inc.

Role- AI Data Engineer

Location- Princeton, NJ & NYC, NY (Hybrid)

Client- Altimetrik / S&P Global

Salary- 160- 170K USD

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

Apply

Vacancy posted 5 hours ago

Similar jobs that could be interesting for youBased on the AI Data Engineer in Princeton, NJ vacancy

AI Data Engineer
...AI Data EngineerThe 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...
Suggested
Reuben Cooley, Inc.
Princeton, NJ
4 days ago
AI & Data Engineer
...development rooted in digital strategy, experience design, and engineering, with a unique combination of agility, scale, and maturity. We... ...management capabilities with hands-on technical proficiency in Python, AI/ML model integration, and prompt engineering to drive next-...
Suggested
Socket.dev
Lawrenceville, NJ
3 days ago
AI/ML & Analytics Platform Engineering in Plainsboro Township, NJ
$110k - $145k
...available Position Overview We are seeking an experienced AI/ML & Analytics Platform Engineer to design, build, and scale enterprise-grade platforms that enable Machine Learning Engineers, ML Researchers, Data Scientists, and Data Analysts to develop, deploy, and...
Suggested
Permanent employment
Local area
Relocation
GD Resources
Plainsboro, NJ
13 days ago
Principal Scientist, Data Science (Translational Knowledge Engineering)
$117k - $201.25k
...employees and recognize their merit.Job Function: Data Analytics & Computational SciencesJob Sub... ...and knowledge architecture that enables AI-driven reasoning across the drug discovery... ..., safety experts, data scientists, AI engineers, and platform teams to create knowledge...
Suggested
Full time
Local area
Immediate start
Johnson & Johnson
Titusville, NJ
1 day ago
AI Engineer
$132k - $142k
...human progress worldwide.POSITION SUMMARY:The Digital Workplace AI Engineer/Developer leads the design and implementation of enterprise AI... ...Copilot Studio agents and connectors for integration with internal data sources and workflows. Develop prompt libraries and role-...
Suggested
Full time
Worldwide
Educational Testing Service
Princeton, NJ
3 days ago
Senior Data Management Professional - Data Engineering - Entities
$110k - $190k
Senior Data Management Professional - Data Engineering - Entities Location Princeton Business Area Data Ref # 10052566 Description &... ...and client needs. You’ll play a key role in shaping how AI, LLMs, rules-based extraction, and workflow automation...
Temporary work
For contractors
Work experience placement
Bloomberg
Princeton, NJ
1 day ago
Data Engineer II
$76.73k - $127.88k
...wants to work in a collaborative environment? As an experienced Data Engineer, you will have the ability to share new ideas and collaborate... ...quality, and driving operational outcomes The TeamAI & Data - AI & Engineering leverages cutting-edge engineering capabilities to...
Local area
Visa sponsorship
Deloitte
Princeton, NJ
1 day ago
Senior Data Management Professional - Data Engineering - Corporate Actions
$110k - $190k
Senior Data Management Professional - Data Engineering - Corporate Actions Location Princeton Business Area Data Ref # 10052844 Description... ...-on data engineering and the practical application of AI/LLMs for automated data extraction, validation and...
Temporary work
For contractors
Work experience placement
Bloomberg
Princeton, NJ
2 days ago
Data Engineer
$130k - $160k
...development of these therapies. Theradex Oncology is an environment for highly motivated individuals who want to make a difference.The Data Engineer will be responsible for management and maintenance of our critical database systems (Oracle + PostgreSQL), while designing and...
Full time
Work at office
Relocation
Flexible hours
Theradex Systems
Princeton, NJ
1 day ago
Onestream Developer Analytics and Insights Engineer II
$82.6k - $162.8k
...business rules (calculations, validations, allocations, and transformations) using OneStream development tools and best practices.Support data management activities including data loads, mappings, transformations, reconciliations, and error handling within OneStream (and/or...
Local area
Remote work
Deloitte
Princeton, NJ
19 hours ago
Sr. Lead Data Engineer (Databricks & Spark)
...NJEmployment Type: C2CIndustry: Computer SoftwareClient: NessContact: Meghana GorusuCompany: SRI Tech SolutionsJob Title: Sr. Lead Data Engineer (Databricks & Spark)Loction: Princeton, NJ (Hybrid)Responsibilities:Demonstrate a strong sense of ownership and responsibility...
Work experience placement
SRI Tech
Princeton, NJ
1 day ago
Sr. Databricks Data Engineer, Vice President
$120k - $202.5k
...challenging new products in the financial space using cutting edge technology.What you will be responsible forAs Sr. Databricks Data Engineer you willDesign, build and test end to end data pipeline including data ingestion (streaming, events, and batch) on cloud-based infrastructure...
Full time
Temporary work
Work at office
Flexible hours
State Street Bank
Princeton, NJ
3 days ago
Senior AI Engineer
...transform ideas into impact by bringing together data, science, technology and human ingenuity... ...: Build, refine, and use ML Engineering platforms and components; develop and implement... ...a strong focus on LLMs and Generative AI. Integrate Azure OpenAI (GPT-4, GPT-4...
Work at office
Local area
Work from home
Flexible hours
ZS Associates
Princeton, NJ
19 hours ago
Data Engineer
...Job Title: Data Engineer Duration: 6+ Months Location: Princeton, NJ (Hybrid) Job Description: The Data Lake team is responsible for data ingestion from internal source systems in batch/real-time modes, curation and governance of the data assets created in the platform...
Tek Ninjas
Princeton, NJ
2 days ago
Data Engineer
...Data Engineer Minimum Mandatory Skills: ADF, Data Bricks Desired Skills: Python, PySpark Job Description: Detailed JD - Key Responsibilities: Design and implement ETL/ELT pipelines using Azure Data Factory Develop and optimize data processing workflows...
E-Solutions
Princeton, NJ
1 day ago
Data Engineer
...Lead Data Engineer/ArchitectLocation: Plainsboro, NJ – (Locals) Need to work 3-4 days onsite per week, mandatoryType: C2CPay rate: $65/hr on C2CKey: Strong Snowflake architectural exposure with ETL, SQL, Datawarehouse and IICs/IDMC. Any Pharma, Healthcare or Life Science...
Local area
Shift work
3 days per week
Argyle Infotech
Plainsboro, NJ
2 days ago
Data Engineer
...Data EngineerDuration: 6+ months Location: 3 days hybrid in Princeton, NJ (local or within drivable distance) MoI: VideoExperience... ...Technologies: Azure SQL, MongoDB, PySparkExperience Required: Data Engineering Experience in implementing Microsoft BI/Azure BI solutions like...
Local area
InterSources
Princeton, NJ
2 days ago
Senior Data Engineer
...Data Engineering Position Experience: 8+ years of dedicated data engineering or data warehousing experience. Snowflake Expertise: Strong command of Snowflake architecture, resource monitors, zero-copy cloning, and time travel. SQL & Programming: Advanced SQL...
ATYETI
Princeton, NJ
1 day ago
Senior Data Engineer
$110k - $190k
...Senior Data Engineer - Financial Markets Princeton, New Jersey Compensation: $110,000-$190,000+ Base Salary (Negotiable) + Annual... ...opportunities to improve automation using Machine Learning and AI Drive continuous improvements across financial data...
Full time
Relocation package
She Recruits LLC
Princeton, NJ
2 days ago
Director, AI Engineering and Enablement
$193.04k - $233.91k
...the careers of those who do it. Position Summary As Director, AI Engineering and Enablement within Bristol Myers Squibb’s AI Engineering and... ...please visit this page for important additional information: Data Protection We will never request payments, financial...
Hourly pay
Full time
Summer work
Live in
Local area
Remote work
Flexible hours
Biopharma Careers
Princeton, NJ
1 day ago
Security Data Solutions Engineer
$120k - $202.5k
Who we are looking forState Street's Cyber Data & Analytics (CyberDNA) team is seeking a Security Data Solutions Engineer to shape the next generation of cybersecurity data, analytics, and AI-powered solutions. Partnering closely with Global Cyber Security teams, Infrastructure...
Full time
Temporary work
Work at office
Local area
Flexible hours
Shift work
2 days per week
State Street Bank
Princeton, NJ
19 hours ago
Director, AI Engineering and Enablement
$193.04k - $233.91k
...Director, AI Engineering And Enablement Working with Us Challenging. Meaningful. Life-changing. Those aren't words that are usually associated with a job. But working at Bristol Myers Squibb is anything but usual. Here, uniquely interesting work happens every day,...
Hourly pay
Full time
Temporary work
Part time
For contractors
Summer work
Work at office
Remote work
Flexible hours
Shift work
Bristol-Myers Squibb Company
Princeton, NJ
19 hours ago
Data Engineer
...sensing algorithms, tools, and techniques to deliver world-class data exploitation capabilities supporting missile defense;... ...requirements will not be considered. SciTec is seeking a Data Engineer to work as part of our Project Control team to design, build, and...
For contractors
Remote work
SciTec
Princeton, NJ
2 days ago
Data Engineer
...Windsor, NJ, USA Employment Type Full-Time About the role The Data Engineer will design and implement cloud migration strategies, leveraging... ...experienced measurable gains in uptime and operational efficiency. AI Capabilities Showcase Don’t hesitate to connect withus! Explore...
Full time
Work experience placement
Remote work
Flexible hours
Quantum Integrators Group
West Windsor, NJ
1 day ago
Principal Data Engineer
...Data EngineerDevelops and maintains scalable data pipelines and builds out new API integrations to support continuing increases in... ...that depend on it.Writes unit/integration tests, contributes to engineering wiki, and documents work.Performs data analysis required to troubleshoot...
ATYETI
Princeton, NJ
2 days ago
Senior AI Application Engineer
...Senior AI Application EngineerWorking with Us Challenging. Meaningful. Life-changing.... ...their personal lives.As a Senior Application Engineer within Bristol Myers Squibb's AI Venture... ...memory, and context services using AWS-aligned data stores such as S3, Athena, PostgreSQL/RDS,...
Bristol-Myers Squibb Company
Princeton, NJ
2 days ago
Senior AI Security Automation Engineer
$120k - $202.5k
Who we are looking forState Street's Cyber Data & Analytics (CyberDNA) team is seeking a Senior AI Security Automation Engineer to help shape the next generation of cybersecurity data, analytics, and AI-powered platforms. Partnering closely with Global Cyber Security teams...
Full time
Temporary work
Work at office
Local area
Flexible hours
Shift work
2 days per week
State Street Bank
Princeton, NJ
3 days ago
Global Head of Data Platform Engineering, SVP
$225k - $337.5k
Who we are looking for Design, build, and operate enterprise-grade, AI-ready data platforms at scale, using modern engineering, Agile, and Site Reliability Engineering (SRE) practices—enabling secure, resilient, and high-performance data capabilities across all State Street...
Full time
Temporary work
Flexible hours
State Street Bank
Princeton, NJ
2 days ago
GCP Data Engineer
...5+ years hands-on Google Cloud Platform (GCP)Strong in BigQuery, Data Fusion, Dataflow, Dataproc, Pub/Sub, Cloud Composer, Cloud StorageDeep... ..., Kafka, Airflow, Terraform, Docker, KubernetesAI/ML or Vertex AI exposurePreferred CertificationsGoogle Professional Cloud...
ASB Resources
Princeton, NJ
2 days ago
Strategic Student Program: AI & Engineering Data Internship (Spring 2027, GSCS)
$18 - $50 per hour
...university graduation and completion of the program Position Overview: Siemens Digital Industries Software is seeking an AI & Engineering Data Intern to support emerging Artificial Intelligence and Digital Thread initiatives within the Capital PreSales organization....
Remote job
Hourly pay
Full time
Internship
Local area
Siemens
Trenton, NJ
1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to AI Data Engineer. Be the first to apply!