Machine Learning Engineer

What You’ll Do

• You will develop scalable data pipelines, optimize models for performance and accuracy, and evaluate them to ensure they are production-ready. 
• Develop, design and implement improvements in user experience in conversational interactions leveraging LLMs in new ways to advance product goals.
• Work with the product team to analyze user behavior and prioritize evolving requirements.
• Experiment at a high velocity, conducting statistical analyses, to optimize the end user experience 
• Research and development on new Conversational AI approaches leveraging cutting edge LLM/NLP advancements.
• Documentation of models, prompts, and processes to increase replicability and drive quality improvements. 
• Stay current with the latest leading research advancements in ML, LLMs, and Conversational AI
• Support other coding and feature development where required

What We’re Looking For

• Bachelor’s degree in Computer Science, Engineering, Mathematics, related field, or equivalent experience
• 3+ years of professional experience (or equivalent) in software engineering, AI/ML development (ideally including a Master’s or Ph.D. in Computer Science, ML, Data Science, or a related field)
• Practical experience and theoretical knowledge of language technologies such as: dialogue/conversational systems, NLP, and Information Retrieval
• Strong foundation in data structures, algorithms, and software engineering principles.
• Proficiency in Python and relevant deep learning frameworks; training (e.g. PyTorch, Tensorflow, JAX, Hugging Face Transformers/Adapters), serving (e.g., Hugging Face TGI//outlines, vLLM)
• Experience with LLM model development and deployment ideally including experience with model distillation, supervised fine-tuning using RLHF/DPO, and automatic prompt tuning (e.g. DSPy, TextGrad) 
• Experience with cloud deployment of ML systems (e.g., AWS, GCP, Azure) including and open systems (e.g. Docker and Kubernetes) and associated ML services.
• Strong analytical and problem-solving skills
• Experience structuring and running data-backed experiments
• Strong written and verbal communication skills
• Exposure to early-stage startups, preferably B2B SaaS

What You’ll Get

• Ownership of projects and strategic priorities regardless of seniority in our learning-focused environment.
• Strong ties to the executive team, a culture of transparency and engagement with strategic decisions.
• Options from day one, which means you will be on the ownership track right away.
• Competitive salary and equity packages.
• Comprehensive health coverage (medical, dental, and vision) from day 1.
• Provision of anything you need to be successful -  learning tools, hardware, office equipment, software.
• 401k optionality for US based employees
• Generous PTO, company-wide R&R shutdowns and paid leave for parents. 
• A WFH stipend, phone stipend and support to work in a We Work or other space as preferred.