C

Caterpillar

Makes heavy machinery, engines and gas turbines

AI Platform Engineer

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeNo salary listedIn-person

About the Job

Requirements

  • A master's degree or higher in Computer Science, Artificial Intelligence, Software Engineering, Data Engineering, Automation, or a related major.
  • Five to eight years of research and development experience in software, data platforms, machine learning platforms, or MLOps, with the ability to independently own the design and delivery of key platform modules.
  • Proficiency in Python and strong software engineering skills, with familiarity in at least one mainstream machine learning framework such as PyTorch or TensorFlow.
  • Familiarity with data processing, distributed training, experiment tracking, Model Registry, model evaluation, and CI/CD in machine learning lifecycles.
  • Familiarity with one or more of Kubernetes, Docker, MLflow, Ray, Airflow, Kubeflow, SageMaker, or similar platforms.
  • Understanding of engineering requirements for data and model versioning, lineage, reproducible training, deployment pipelines, and production monitoring.
  • Ability to communicate and analyze problems across teams, and communicate about technical topics in English.

Responsibilities

  • Build and operate scalable AI data, training, evaluation, and MLOps infrastructure to support teams developing, validating, and managing AI models, while ensuring data, models, configurations, and evaluation results can be reused, traced, and continuously evolved.
  • Build collection, ingestion, processing, and storage capabilities for autonomous vehicle and mining-site data, supporting unified management of sensor data, vehicle status, actions, operational data, and simulation data.
  • Establish dataset versioning, metadata, annotation workflows, and data-quality mechanisms to support Corner Case, Failure Case, Demonstration Data, and multimodal Vision-Language-Action data management.
  • Build reusable training tasks, compute scheduling, experiment tracking, and automated evaluation capabilities for traditional machine learning, deep learning, VLM/VLA, and back-office optimization models.
  • Build a Model Registry, artifact management, and model lifecycle processes, maintaining traceability among code, data, configuration, checkpoints, models, and evaluation results.
  • Provide model packaging, candidate-version management, deployment-pipeline infrastructure, and rollback mechanisms; model release decisions and release scope remain the responsibility of the relevant On-board or Back-office product teams.
  • Improve the stability, scalability, resource utilization, and developer experience of data and training pipelines, and establish monitoring, alerting, and diagnostic capabilities.
  • Collaborate with perception, prediction, planning, VLA, scheduling optimization, AI Runtime, Validation, Cybersecurity, and Functional Safety teams to turn shared needs into standardized platform capabilities.

Desired Qualifications

  • Experience with data and MLOps platforms for autonomous driving, robotics, industrial AI, or off-highway environments such as mining.
  • Experience with VLM/VLA multimodal data, Demonstration Data, simulation data, or reinforcement-learning training platforms.
  • Experience with large-scale sensor data, annotation platforms, data quality, or data governance.
  • Experience with LLM/LLMOps, model post-training, or AI governance.

About the Company

Caterpillar

Makes heavy machinery, engines and gas turbines

Caterpillar makes construction and mining machines, diesel and natural gas engines, industrial gas turbines and locomotives for contractors, mines, energy companies and data center builders. It is the world's largest maker of construction and mining equipment, and its yellow Cat bulldozers and excavators work in nearly every country. Customers buy or rent machines through independent Cat dealers, while its engines and turbines generate electricity and push natural gas through pipelines. Caterpillar earns money from equipment sales and from the parts and repairs that dealers supply over a machine's working life. It was formed in 1925 when California's Holt and Best tractor companies merged, and it is based in Irving, Texas.

Company size
10,001+
Company stage
IPO
Headquarters
Irving, TX
Founded
1925

Recently Posted Jobs

Explore more opportunities at Caterpillar

See more jobs