Founding Machine Learning Engineer

About this role

A Series A AI/ML startup is seeking a Founding Machine Learning Engineer to build production-grade ML systems from the ground up, focusing on model training, optimization, and deployment. You'll work directly with the founding team to design end-to-end pipelines, implement LLMs and generative models, and establish technical infrastructure and best practices for a rapidly scaling organization.

What you'll do

• Design and optimize end-to-end ML pipelines from data ingestion through model deployment
• Implement and fine-tune large language models, embeddings, and generative models for production use
• Develop distributed training and inference systems for scalable ML workloads
• Build model monitoring, evaluation, and continuous learning frameworks
• Establish best practices for model versioning, reproducibility, and infrastructure
• Partner with data and product teams to translate research ideas into measurable ML impact

What they're looking for

• Python
• PyTorch, TensorFlow, or JAX
• End-to-end ML pipeline development
• LLM and generative model fine-tuning
• Distributed training and inference systems
• AWS, GCP, or Azure
• MLflow or Weights & Biases
• Cross-functional collaboration

Benefits

• Compensation: $220,000–$300,000 USD annually
• Founding team role with significant technical influence
• Remote-friendly with Mountain View office
• Opportunity to shape ML culture and infrastructure from scratch
• Work on cutting-edge generative AI and model training