Senior Machine Learning Engineer

Posted:
8/21/2024, 11:14:09 AM

Location(s):
San Francisco, California, United States ⋅ California, United States

Experience Level(s):
Senior

Field(s):
AI & Machine Learning ⋅ Software Engineering

Workplace Type:
Hybrid

Pay:
$96/hr or $199,680 total comp

Who We Are 

Moveworks is the universal AI copilot for search and automation across all your business applications. We give employees one place to go to find information and get support while reducing costs for your business. The Moveworks Copilot is powered by an industry-leading Reasoning Engine that uses a combination of public and proprietary language models to understand employee queries, then build and execute multi-step plans that achieve them. It does this by linking into systems (like ITSM, HRIS, ERP, identity management, and more) with native and custom-built integrations that turn natural language into powerful automations for employees.  

The world’s most innovative brands like Databricks, Broadcom, Hearst, and Palo Alto Networks trust Moveworks to eliminate repetitive support issues, deliver instant knowledge, and empower employees to work faster across applications.

Founded in 2016, Moveworks has raised $315 million in funding, at a valuation of $2.1 billion, thanks to our award-winning product and team. In 2023, we were included in the Forbes Cloud 100 list as well as the Forbes AI 50 for the fifth consecutive year. We were also recognized by the 2023 Edison Awards for AI Optimized Productivity, and were included on Fast Company's Most Innovative Companies list for 2024! 

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

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

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

    Compensation: $129,000-$257,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.