Senior Machine Learning Engineer

What You Will Do

We are looking for a Senior Machine Learning Engineer to join us in expanding Moveworks NLU capabilities, crystallizing enterprise text into rich and actionable representations to enable magical user experiences and improving Moveworks generative and conversational AI capabilities platform-wide.

As a member of the NLU team, you will have all the tools of modern NLP and NLG at your disposal, from best-in-class LLMs, MLMs, and hybrid vector databases to all the infrastructure needed to fine-tune, evaluate, and serve your own models in production. We are a data-centric team, and you will have the assistance of a world-class annotation team to build error-free, inclusive, and privacy-preserving datasets for model training and evaluation. Your work will impact our team’s core objective to understand every enterprise issue, in deep collaboration with other functions within Moveworks.

If you are looking to do the best work of your career with a bright, dedicated, and impact-driven team alongside you, we’d love to have a conversation.

• Apply machine learning and software engineering to create lasting value for all our customers
• Take on exciting and difficult challenges in NLU, LLM, and conversational agent domains, such as coreference-preserving data masking, RLHF/RLAIF/DPO, fine-tuning LLMs for tool use and enterprise reasoning, machine translation, speech recognition, agent evaluation, active learning of exemplars for few-shot text classification, abstractive summarization, and grounding & verifiability for generated text.
• Push the envelope of Moveworks commitments to responsible AI, expanding our infrastructure for ensuring models work equally well for all people, red-teaming models to ensure they behave safely and as intended, and keeping our ML at the cutting edge of data privacy and security
• Use your knowledge of machine learning fundamentals to design new algorithms and architectures, evaluate them with small scale experiments and productionize your solutions at scale
• Research and develop innovative, scalable and dynamic solutions to hard problems
• Use the latest advances in machine learning and LLMs to enhance our products and create delightful user experiences
• Spend time weekly reading, discussing, and potentially building models out of the latest ML research and open-source code

What You Bring To The Table

• Ability to solve problems end-to-end with machine learning, including dataset curation, model training, evaluation, and deployment
• Drive to ship product improvements with production-quality, fully unit-tested code and rigorously-evaluated model updates
• Solid grasp of model evaluation fundamentals, especially for text classification and non-uniform sampling regimes
• Working familiarity with machine learning frameworks such as PyTorch and LightGBM
• Comfort developing on Mac and Linux with languages like Python, C++, and/or Go
• Knowledge of deep learning architectures and algorithms and leading large language models
• Attention to detail and high standard of data quality for training and especially evaluation datasets
• Desire to work at a startup pace in a medium-sized company with a high degree of ownership
• Strong appetite for continuous incremental wins and completing challenging projects fast
• High level of curiosity about engineering outside of immediate discipline and ongoing desire to learn and stay at the cutting edge of NLU & AI

Nice To Haves

• Experience productionizing ML models at scale
• Experience in AI fairness, privacy, safety, and/or security
• Experience with NLP libraries such as HuggingFace Transformers, PEFT, and SpaCy 3
• Experience iterating on prompts for large language models in a data-driven way

Compensation Range: $193,000 - $257,000