Principal AI Engineer
šŗšø United States
Management
Machine Learning
Design
Large Language Models
Sales
Data Science
Testing
$159,600.00 - $207,500.00
Principal AI Engineer
from šŗšø United States
$159,600.00 - $207,500.00
Job Description:
The Principal Engineer, AI Model Training & Data Strategy owns how Commercial AI (CAI) products train, fine-tune, and evaluate models, and how the data behind those models is sourced, curated, stored, and governed. This is primarily a model-training role with a strong secondary focus onĀ the dataĀ management and pipelines that make high-quality training possible. The role defines the enterprise training strategy and the standards for how and where training data from Commercial AI products is stored, versioned, and reused.Ā
Essential Job Functions and ResponsibilitiesĀ
Define and own the end-to-end model training strategy across CAI products, spanning traditional AI/ML models and large language modelsĀ
Fine-tune large language models using parameter-efficient techniques (e.g.,Ā QLoRA,Ā LoRA, PEFT) and full fine-tuningĀ whereĀ warrantedĀ
Train, evaluate, and tune traditional AI/ML models (classification, regression, ranking, clustering, and similar)Ā
Work with large volumes of data ā design andĀ optimizeĀ pipelines for ingestion, cleaning, labeling, and feature engineeringĀ
Define standards for how and where training data from Commercial AI products is stored, versioned, and accessed (data lakes/warehouses, feature stores, dataset registries)Ā
Establish data governance, lineage, quality, licensing/consent, and PII-handling practices for training dataĀ
Build reproducible training pipelines and experiment tracking (datasets, hyperparameters, checkpoints, and metrics)Ā
Define evaluationĀ methodologyĀ and benchmarks for model quality, including offline evaluation and regression testingĀ
Curate and clean training, validation, and test datasets, including synthetic data generation where appropriateĀ
OptimizeĀ training cost and computeĀ utilizationĀ (GPU efficiency, distributed training, quantization)Ā
Partner with product and platform teams to operationalize and hand off trained and fine-tuned models to productionĀ
Mentor engineers and raise model-training and data-quality maturity across teamsĀ
Knowledge, Skills, and AbilitiesĀ
Strong hands-on experience training and fine-tuning both traditional AI/ML models and LLMs in productionĀ
Deep experience with parameter-efficient fine-tuning (QLoRA,Ā LoRA, PEFT), quantization, and the tradeoffs versus full fine-tuningĀ
ProficiencyĀ with ML/DL frameworks and libraries (e.g.,Ā PyTorch, Hugging Face Transformers/PEFT/TRL, scikit-learn)Ā
Experience building andĀ operatingĀ large-scale data pipelines and platforms (e.g., Spark, Ray,Ā dbt, or equivalents)Ā
Strong grasp of data management: dataset storage architecture, versioning, lineage, governance, and PII handlingĀ
Experience with experiment tracking and reproducible ML (e.g.,Ā MLflow, Weights & Biases)Ā
Understanding of distributed training and GPU/compute optimizationĀ
Ability to define strategy and standards whileĀ remainingĀ hands-on in codeĀ
Strong stakeholder collaboration and problem-solving skillsĀ
Education and ExperienceĀ
Bachelorās degree in Computer Science, Engineering, or related discipline; advanced degree in ML, AI, or Data Science preferredĀ
12 or more years of experience in AI/ML engineering, applied ML, or data engineering, with significant hands-on model training and fine-tuningĀ
DisclaimerĀ
The above statements describe the general nature and level of work performed in this role. Other duties may be assigned.Ā
Vertex Values: Together We Win
We're building a team of people who are passionate about making an impact for our customers and committed to how that impact is achieved. Our values define the behaviors, mindset, and culture that make Vertex a great place to grow and do meaningful work.
Play to Win or We Don't Play ā If we choose to do something, we're choosing to do it because we plan to win. That mindset raises our bar on product quality, customer outcomes, and how we show up for one another.
Work As a Team, Putting the Customer At the Core ā Our customers are our true north. Whatever your role, ask: how will this help a customer succeed today? We earn trust through outcomes, not promises.
Achieve Excellence With Integrity, Speed, and Agility ā The market isn't slowing down. We'll move faster, adapt quickly, and never compromise on doing things the right way ā for teammates, customers, and partners.
Innovate Boldly With a Growth Mindset ā Progress demands smart risk. We'll try new approaches, learn fast, and keep pushing the boundaries ā especially where AI can remove friction and unlock value.
Communicate with Care, Candor and Transparency ā Honest, constructive conversations make us better. Let's speak plainly about what's working and what isn't and help each other improve.
Pay Transparency Statement:
Base pay offered to new hires may vary based upon factors including relevant industry and job-related skills and experience, geographic location, and business needs.* The range displayed does not encompass the full potential of the role, which allows for further growth and career progression.
In addition, as a part of our total compensation package, this role may be eligible for the Vertex Bonus Plan (VOB), a role-specific sales commission/bonus, and/or equity grants.
Learn more aboutLife at Vertex and connect with your recruiter for more details regarding Vertex's compensation and benefit programs.
*In no case will your pay fall below applicable local minimum wage requirements.