AI Platform Engineer
Caterpillar2 days ago
Wuxi, China or Shanghai, ChinaSenior / Staff+
Responsibilities
- Build and operate data platforms for collecting, ingesting, processing, and storing autonomous-driving vehicle and mining-site data.
- Manage dataset versions, metadata, annotation workflows, data quality, corner cases, failure cases, demonstration data, and multimodal Vision-Language-Action data.
- Develop reusable training jobs, compute scheduling, experiment tracking, and automated model evaluation capabilities.
- Build MLOps lifecycle processes covering model registries, artifact management, traceability, packaging, candidate versions, deployment pipelines, and rollback.
- Improve data and training pipeline reliability, scalability, resource utilization, developer experience, monitoring, alerting, and issue diagnosis.
- Collaborate with perception, prediction, planning, VLA, scheduling optimization, AI Runtime, Validation, Cybersecurity, and Functional Safety teams.
Requirements
- Master’s degree or higher in computer science, artificial intelligence, software engineering, data engineering, automation, or a related field.
- 5–8 years of software, data-platform, machine-learning-platform, or MLOps development experience, with the ability to independently design and deliver critical platform modules.
- Proficiency in Python and strong software engineering skills.
- Familiarity with at least one major machine-learning framework, such as PyTorch or TensorFlow.
- Knowledge of data processing, distributed training, experiment tracking, model registries, model evaluation, and machine-learning lifecycle capabilities.
- Familiarity with one or more of Kubernetes, Docker, MLflow, Ray, Airflow, Kubeflow, and SageMaker, or comparable platforms.
- Understanding of data and model versioning, lineage, reproducible training, deployment pipelines, and production monitoring.
- Strong cross-team communication, problem analysis, and English technical communication skills.
- Experience with autonomous driving, robotics, industrial AI, mining, multimodal VLM/VLA data, demonstration data, simulation data, reinforcement-learning training platforms, large-scale sensor data, annotation platforms, data quality, data governance, LLMOps, model post-training, or AI governance is preferred.
Benefits
- Posting period is November 4, 2026 through November 19, 2026.
- Caterpillar is an equal opportunity employer and encourages qualified applicants of any age to apply.
Tech Stack
Categories
Data EngineeringML Engineering
About Caterpillar
Caterpillar Inc. designs, manufactures, and services construction and mining equipment, diesel and natural gas engines, industrial gas turbines, and diesel-electric locomotives for infrastructure, resource, and energy customers. The public company, founded in 1925 and headquartered in Irving, Texas, sells primarily through an independent worldwide dealer network and provides equipment financing. It operates through Construction Industries, Resource Industries, and Energy & Transportation segments, with shares traded on the NYSE and Euronext Paris.