Machine Learning Engineer - LLM

$160,000 - $190,000/Yr

Moveworks - Mountain View, CA

Full-time - Mid Level

Mountain View, CA

251-500 employees

Truck Transportation

About the position

The Machine Learning Engineer - LLM at Moveworks is responsible for designing, building, and optimizing scalable machine learning infrastructure to support the training, evaluation, and deployment of large language models. This role involves collaborating with cross-functional teams to develop new features and drive best practices in machine learning and data engineering.

Responsibilities

• Design, build and optimize scalable machine learning infrastructure for large language models.
• Build abstractions to automate various steps in different ML workflows.
• Collaborate with cross-functional teams of engineers, data analytics, machine learning experts, and product to build new features.
• Drive best practices in machine learning and data engineering.

Requirements

• Significant academic background in ML and/or industry experience in Machine Learning, Infrastructure or related fields.
• Expertise in Python and Golang.
• Experience with building end-to-end machine learning systems.
• Experience building scalable microservices and ETL pipelines.
• Experience with deep learning frameworks such as Pytorch or Huggingface.
• Effective communicator with experience collaborating cross-functionally.

Nice-to-haves

• Experience with ML Inference frameworks such as TensorrtLLM or Nvidia Triton.
• Experience with distributed training frameworks such as Deepspeed.

Benefits

• Market competitive salary
• Equity for all full-time roles
• Exceptional benefits
• Commissions or bonus plans for applicable roles