Senior Machine Learning Engineer II, NLU & Agentic AI

The Role

We are looking for an experienced software engineer with machine learning expertise to join us in expanding Moveworks NLU (natural language understanding) and agentic AI capabilities, enabling increasingly 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, multimodal foundation models, 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.

You will also go beyond model training to achieve state-of-the-art AI performance in production, in every meaning of the word “performance”: not just accuracy and quality of outputs, but also latency, reliability, and capability of the end-to-end user-facing system as a whole. Successful machine learning engineers on the team are just as motivated to design and evolve great compound AI systems and their components as they are to train great models.

Our team indexes on increasing our ability to move fast, solving challenging product and engineering challenges, and pushing the envelope of value provided to customers. Your work will impact our team’s core objectives to understand every enterprise issue and build the most reliable copilot the world has ever seen, in deep collaboration with other functions within Moveworks.

What You Will Do

• Apply software engineering, machine learning, and compound AI system engineering to create lasting value for all our customers

• Take on exciting and difficult challenges in conversational agent domains, such as agent cognitive architecture iteration, multimodal agents, multilingual agents, conversational memory management, reasoning strategies (eg Tree of Thoughts / Graph of Thoughts), fine-tuning LLMs for tool use and enterprise reasoning (including preference alignment with RLHF/RLAIF/DPO), 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 and LLMs 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

• Drive to ship product improvements with production-quality, fully unit-tested code and rigorously-evaluated updates to models, prompts, or other tunable system components

• Ability to solve problems end-to-end with machine learning

• Solid grasp of model evaluation fundamentals, especially for text generation, text classification, and non-uniform sampling regimes

• Attention to detail and high standard of data quality for training and especially evaluation datasets

• Readiness to hit the ground running in a Mac development environment, programming in Python and/or Golang

• Knowledge of deep learning architectures and algorithms and leading large language models

• Desire to work at a startup pace in a medium-sized company with a high degree of ownership

• Drive to ship product improvements with production-grade code

• 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 Have

• Experience productionizing ML models at scale

• Experience in AI fairness, privacy, permission controls, safety, and/or security

• Working familiarity with machine learning frameworks such as PyTorch and LightGBM

• 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

Base salary compensation range: $150,000-$275,000

*Our compensation package includes a market competitive salary, equity for all full time roles, exceptional benefits, and, for applicable roles, commissions or bonus plans.
Ultimately, in determining pay, final offers may vary from the amount listed based on geography, the role’s scope and complexity, the candidate’s experience and expertise, and other factors.

Moveworks Is An Equal Opportunity Employer
*Moveworks is proud to be an equal opportunity employer. We provide employment opportunities without regard to age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, veteran status, or any other characteristics protected by law.

Who We Are

Moveworks is an AI Assistant that helps all employees find information, automate tasks, and be more productive. We give the entire workforce one interface to get answers and take action across every enterprise system. And for developers, we make it easy to build and deploy AI agents that bring the power of Moveworks to every business process or workflow.

It’s all powered by a pioneering Reasoning Engine paired with an Agentic Automation Engine that, together, are able to handle even the most complex requests by understanding queries, then building and executing intelligent plans to fulfill them — in seconds.

Founded in 2016, Moveworks has raised $315M in funding, and eclipsed $100M in ARR in 2024 thanks to our award-winning product and team. Along the way, we’ve earned recognition as a leader in the Forrester Wave for Conversational AI Platforms for Employee Services, as a member of the Forbes Cloud 100 and AI 50 lists, and as one of America’s Most Loved Workplaces according to Newsweek.

Today, Moveworks has over 500 employees in six offices globally, and is backed by some of the world's most prominent investors including Kleiner Perkins, Lightspeed, Bain Capital Ventures, Sapphire Ventures, Iconiq, and more.

Over 350 leading organizations like Marriott, Databricks, Toyota, CVS Health, and Honeywell trust Moveworks to increase operational efficiency, enhance the employee experience, and drive lasting AI transformation.

Come join one of the most innovative teams on the planet!