Machine Learning Engineer

We are seeking a Machine Learning engineer to assist in enhancing Moveworks NLU capabilities, transforming complex enterprise text into clear, actionable representations to provide exceptional user experiences and improve the Moveworks generative and conversational AI functionality across the platform.

In the NLU team, you'll have modern NLP and NLG tools at your disposal, ranging from top-of-the-line LLMs, MLMs, and hybrid vector databases to the infrastructure necessary to fine-tune, evaluate, and utilize your models in production. As a data-centric team, you'll receive support from a world-class annotation team to develop accurate, inclusive, and privacy-respecting datasets for model training and assessment. Your efforts will significantly contribute to our core aim of comprehending every enterprise issue, working closely with other departments within Moveworks.

If you're ready to produce your career's best work alongside a talented, dedicated, and impact-focused team, we're eager to start a conversation.

About us:

Moveworks is committed to making language the universal UI. We provide enterprises with a conversational interface that works across every system — from Microsoft to Workday to Salesforce. Utilizing GPT-class machine learning models, Moveworks' enterprise copilot learns the unique language of each organization to tackle thousands of use cases. Brands like Databricks, Broadcom, DocuSign, and Palo Alto Networks use Moveworks' proprietary enterprise data, ready-to-use solutions, and straightforward developer tools to incorporate conversational automation into all areas of their business.

Established in 2016, Moveworks has raised $315 million in funding, with a valuation of $2.1 billion. We have been included in the 2023 Forbes Cloud 100 list and the Forbes AI 50 list for five consecutive years, and have been recognized as the 2023 Edison Awards winner for AI Optimized Productivity, as well as the Best Bot Solution at the 2022 AI Breakthrough Awards.

With over 500 employees in six offices globally, Moveworks is backed by a group of the world's foremost investors, such as Kleiner Perkins, Lightspeed, Bain Capital Ventures, Sapphire Ventures, Iconiq, and others.

Join one of the world's most innovative teams!

Your responsibilities:

• Apply machine learning and software engineering concepts to create lasting value for our customers
• Address challenging problems in NLU, LLM, and conversational agent domains, such as data masking that preserves coreference, 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.
• Expand Moveworks' dedication to responsible AI by enhancing our infrastructure to ensure models work effectively for everyone, by testing models to ensure they operate safely and as designed, and by keeping our ML at the forefront of data privacy and security.
• Utilize your understanding of machine learning principles to create new algorithms and architectures, evaluate them through small scale experiments, and implement your solutions on a larger scale.
• Research and develop innovative, scalable, and dynamic solutions to hard problems.
• Employ the most recent advancements in machine learning and LLMs to improve our products and create outstanding user experiences.
• Spend time weekly reading, discussing, and possibly creating models from the latest ML research and open-source code.

What you'll need:

• Ability to solve problems from start to finish with machine learning, including dataset preparation, model training, evaluation, and deployment.
• Determination to ship product enhancements with production-grade, fully unit-tested code and rigorously-evaluated model updates.
• Solid understanding of model evaluation basics, particularly for text classification and non-uniform sampling regimes.
• Working knowledge with machine learning frameworks such as PyTorch and LightGBM.
• Comfort in developing on Mac and Linux with languages like Python, C++, and/or Go.
• Knowledge of deep learning architectures, algorithms, and leading large language models.
• Attention to detail and high standard for data quality for training and especially evaluation datasets.
• Desire to work at a startup speed in a medium-sized company with a high degree of ownership.
• Strong drive for continuous incremental wins and quick completion of challenging projects.
• High level of curiosity about engineering outside of immediate discipline and an ongoing desire to learn and stay at the cutting edge of NLU & AI.

It would be nice if you have:

• Experience in scaling ML models in production.
• Experience in AI fairness, privacy, safety, and/or security.
• Experience with NLP libraries like HuggingFace Transformers, PEFT, and SpaCy 3.
• Experience in iteratively improving prompts for large language models in a data-driven way.

Base Compensation Range:  $129,000 - $221,000

*Our total compensation package includes a competitive salary, equity for all full-time roles, superb benefits, and, for relevant roles, commissions or bonus plans. In determining pay, final offers may differ from the amount specified 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 offer employment opportunities irrespective of age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, veteran status, or any other characteristics protected by law.