Note: The job is a remote job and is open to candidates in USA. Maxonic Inc. is a recruiting firm connecting candidates with client opportunities. The company is seeking a Principal Engineer to own enterprise AI model training strategy, fine-tune and evaluate traditional AI/ML models and large language models, and establish standards for training data management. The role also involves building reproducible pipelines, defining model evaluation practices, governing training data, and optimizing compute utilization.
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)
Skills
- • 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
- • Bachelor's degree in Computer Science, Engineering, or related discipline
- • 12 or more years of experience in AI/ML engineering, applied ML, or data engineering, with significant hands-on model training and fine-tuning
- Advanced degree in ML, AI, or Data Science preferred
Benefits
- Remote work arrangement
Company Overview
Company H1B Sponsorship