Job Details

Senior Machine Learning Engineer, NLU & Agentic AI

Machine Learning | San Francisco, CA | Full Time

As a member of the NLU team, you will have access to modern NLP and NLG tools, including best-in-class LLMs, multimodal foundation models, and hybrid vector databases. You will also work with infrastructure to fine-tune, evaluate, and deploy models in production. Our data-centric team provides support through a world-class annotation team to build high-quality, inclusive, and privacy-preserving datasets.

You will go beyond model training to achieve state-of-the-art AI performance, focusing on accuracy, latency, reliability, and overall system capability. Success involves designing and evolving complex AI systems and their components, not just training models.

Our team emphasizes speed, solving challenging engineering problems, and delivering value to customers. Your work will directly impact our goal to understand enterprise issues and develop the most reliable copilot, in close collaboration with other teams at Moveworks.

1. Address challenging problems in conversational agents, including cognitive architecture, multimodal and multilingual agents, memory management, reasoning strategies, and fine-tuning LLMs for enterprise reasoning and tool use.
2. Advance our responsible AI initiatives by expanding infrastructure for fairness, safety, privacy, and security.
3. Design new algorithms and architectures, evaluate through experiments, and deploy solutions at scale.
4. Research and develop scalable, innovative solutions to complex problems.
5. Leverage latest ML and LLM advances to enhance products and user experiences.
6. Engage with current ML research and open-source models regularly.

• Commitment to shipping product improvements with high-quality, tested code and evaluated model updates.
• End-to-end problem-solving skills in machine learning.
• Strong understanding of model evaluation, especially for text generation and classification.
• Attention to data quality for training and evaluation datasets.
• Proficiency in Python and/or Golang in a Mac environment.
• Knowledge of deep learning architectures and large language models.
• Ability to work at a startup pace with ownership mindset.
• Drive for incremental wins and tackling challenging projects quickly.
• Curiosity and desire to stay at the forefront of NLU & AI.

• Experience deploying ML models at scale.
• Knowledge of AI fairness, privacy, safety, and security.
• Familiarity with frameworks like PyTorch and LightGBM.
• Experience with NLP libraries such as HuggingFace Transformers, PEFT, and SpaCy 3.
• Prompt engineering for large language models.

Salary Range: $150,000 - $275,000

Join Moveworks, a global company with over 500 employees, backed by top investors, and committed to innovation. Apply now after verifying your identity.