Lead AI Automation Engineer - Edge AI & Data Pipelines

Socket.dev; Irving (TX); On-site; USD 128,000 - 209,000; Full time

Benefits offered by this job

Medical, dental, and vision benefits

Paid time off

401(k) savings plans

Health Savings Account

Job summary

Caterpillar is hiring a Lead AI Automation Engineer for onsite work at the Irving, TX office. You will lead a scrum team focused on annotation automation and AI-driven data production for robotics, autonomy, computer vision, and VLA systems.

You will design and deploy automated labeling and captioning solutions, train AI models for auto-labeling and auto-captioning, and develop scalable data pipelines with a focus on reliability and quality across heavy equipment datasets.

Qualifications

• Bachelor's degree in Computer Science, Software Engineering, AI, ML, Robotics, or a related field.
• Experience leading technical delivery for software or AI engineering teams.
• Experience developing production-grade software solutions using Python and modern software practices.
• Experience designing and implementing distributed systems, APIs, and data pipelines.
• Experience training, fine-tuning, or deploying machine learning models.
• Strong problem-solving skills and ability to work across disciplines.
• 3+ years generating and managing camera and LiDAR datasets for perception or ML apps.
• 3+ years experience in data annotation and QA for autonomous systems or CV projects.
• Experience using annotation and dataset tools like Labelbox, Supervisely, RoboFlow.
• Experience with ML frameworks such as PyTorch, TensorFlow, ONNX, Weights & Biases, MLflow.

Responsibilities

• Lead the technical execution of a scrum team focused on annotation automation and AI-driven data production.
• Design, develop, and deploy automated labeling and captioning solutions for perception and VLA datasets.
• Train, fine-tune, and evaluate AI models to support auto-labeling and auto-captioning.
• Build data pipelines moving datasets through labeling, review, validation, and publication.
• Implement and extend platforms such as NVIDIA Cosmos Curator and Cosmos Reason.
• Develop solutions that integrate Caterpillar data systems with external annotation providers and human-in-the-loop workflows.
• Drive reliability, scalability, observability, and operational excellence across annotation automation services.
• Mentor engineers and provide day-to-day technical leadership for the team.

Skills

Leadership experience

Python

Distributed systems

ML models

Problem solving

Camera/LiDAR data pipelines

Data annotation quality

Annotation tools

ML frameworks

Education

Bachelor's degree in CS/SE/AI/Robotics

Tools

Labelbox

Supervisely

RoboFlow

PyTorch

TensorFlow

ONNX

Weights & Biases

MLflow

NVIDIA Cosmos Curator

Cosmos Reason

Job description

Caterpillar is hiring a Lead AI Automation Engineer for onsite work at the Irving, TX office. You will lead a scrum team focused on annotation automation and AI-driven data production for robotics, autonomy, computer vision, and VLA systems.

You will design and deploy automated labeling and captioning solutions, train AI models for auto-labeling and auto-captioning, and develop scalable data pipelines with a focus on reliability and quality across heavy equipment datasets.