AIML Data Engineer

Socket.dev; Austin (TX); On-site; USD 120,000 - 160,000; Full time

Job summary

Socket.dev is seeking an AIML Data Engineer to own end-to-end data pipelines and telemetry infrastructure supporting AI/ML initiatives in a fast-paced environment. You will transform raw data from conversations into model-ready formats, orchestrate scoring workflows, and partner with program managers to turn data into actionable insights for leadership.

The role blends full-stack data engineering and analytics, requiring strong SQL/Python skills, cloud platform experience, and a track record of

Qualifications

• Bachelor's degree in Computer Science, Data Science, Statistics, Engineering, or related field (or equivalent experience)
• 4+ years of experience in data engineering or analytics engineering
• Strong proficiency in SQL and Python for both data engineering and analytical investigation
• Experience with cloud data platforms (Snowflake, Databricks, or similar)
• Experience with ETL/ELT tools and pipeline orchestration (dbt, Airflow, Prefect, or similar)
• Demonstrated ability to conduct root cause analysis and translate data findings into actionable recommendations
• Experience with version control (Git) and CI/CD practices

Responsibilities

• Design and maintain data pipelines to ingest and transform support conversation data
• Develop feature engineering and telemetry infrastructure for AIML initiatives
• Collaborate with program managers to investigate trends and surface actionable insights
• Provide data-driven deep dives and storytelling to leadership and stakeholders

Skills

SQL

Python

Root cause analysis

Git & CI/CD

Data engineering

Education

Bachelor's degree in CS/Data Science/Engineering

Tools

Snowflake

Databricks

dbt

Airflow

Prefect

Tableau

Streamlit

Job description

Are you passionate about building the data infrastructure that powers AI-driven customer experience measurement — and using that data to uncover the "why" behind the numbers? The AppleCare Customer Insights (ACCI) team is redefining how Apple measures and improves generative support experiences. Our AIML initiatives use large language models to evaluate support conversations across multiple quality dimensions, providing real-time signal to leadership on how our AI-powered support is performing. We are seeking an AIML Data Engineer to own the data pipelines, feature engineering, and telemetry infrastructure that underpin our AIML portfolio — while also serving as a hands-on analytics partner who conducts root cause analysis, targeted investigations, and data-driven deep dives that translate pipeline outputs into actionable insights for program managers and leadership.

Description

The AIML Data Engineer builds and maintains the data foundation that powers ACCI's AI/ML initiatives, and turns that foundation into insight. You will design and implement pipelines that ingest support conversation data, transform it into model-ready formats, orchestrate scoring workflows, and deliver telemetry — then go a step further by partnering with program managers to investigate trends, diagnose performance shifts, and surface the stories in the data that drive decisions. This is a full-stack engineering-and-analytics role that consolidates data pipeline orchestration, model feature engineering, telemetry analytics, and investigative analysis into a single high-impact position.

Minimum Qualifications

• Bachelor's degree in Computer Science, Data Science, Statistics, Engineering, or related field (or equivalent experience)
• 4+ years of experience in data engineering or analytics engineering
• Strong proficiency in SQL and Python for both data engineering and analytical investigation
• Experience with cloud data platforms (Snowflake, Databricks, or similar)
• Experience with ETL/ELT tools and pipeline orchestration (dbt, Airflow, Prefect, or similar)
• Demonstrated ability to conduct root cause analysis and translate data findings into actionable recommendations
• Experience with version control (Git) and CI/CD practices

Preferred Qualifications

• Experience building data pipelines supporting LLM-based systems (RAG, scoring, evaluation)
• Experience with data visualization and storytelling (Tableau, Streamlit, or similar)
• Familiarity with NLP data preparation — tokenization, embedding generation, prompt engineering data flows
• Experience with streaming or event-driven data architectures (Kafka or similar)
• Experience with data quality and observability tools (Great Expectations, Monte Carlo, or similar)
• Understanding of concept drift detection and model monitoring pipelines
• Experience supporting program or product teams with investigative analytics in a customer experience domain