Lead Machine Learning Engineer - AI Agents

3 Months ago • All levels • Research Development • $175,500 PA - $277,500 PA

Job Summary

Job Description

The Lead Machine Learning Engineer will be responsible for designing and delivering infrastructure and models for agentic LLM workflows. They will work on cutting-edge machine learning initiatives, mentor engineers, and contribute to the broader AI research community. This role involves architecting and implementing infrastructure for agent-based LLM systems, training and fine-tuning custom models, leading cross-functional collaboration, and translating experimental ideas into scalable production systems. The engineer will also contribute to Upwork’s visibility in the broader AI/ML research community.
Must have:
  • Proficiency in Python and ML frameworks.
  • Deep understanding of core ML concepts.
  • Experience training large models on GPU clusters.
Good to have:
  • Experience in R&D environments.
  • Publications in major AI/ML conferences.
Perks:
  • Comprehensive medical coverage
  • Unlimited PTO
  • 401(k) plan with matching
  • 12 weeks of paid parental leave
  • Employee Stock Purchase Plan

Job Details

Upwork ($UPWK) is the world’s largest work marketplace, connecting businesses with highly skilled professionals worldwide. From entrepreneurs to Fortune 100 enterprises, companies trust Upwork’s platform to access expert talent, leverage AI-powered work solutions, and drive meaningful business outcomes.

Upwork’s AI-powered platform has facilitated over $20 billion in economic opportunity for professionals worldwide. With professionals spanning 10,000+ skills, including AI and machine learning, software development, sales and marketing, customer support, finance and accounting, and more, Upwork empowers businesses of all sizes to scale, innovate, and build agile teams.


We’re looking for a Lead AI/ML Engineer to help shape the future of intelligent systems at Upwork. In this role, you’ll design and deliver cutting-edge infrastructure and models that power agentic LLM workflows across our platform. You’ll work at the intersection of research and production, owning the delivery of advanced ML solutions and driving business outcomes through applied innovation. 

This role offers the chance to lead high-impact projects, support and mentor others, and help shape the future of autonomous AI agents globally. You'll work on cutting-edge machine learning initiatives that drive real business results while contributing to the broader AI research community through publications at leading conferences.

Responsibilities

  • Architect and implement core infrastructure to support agent-based LLM systems, including multi-agent pipelines, RAG, and real-time orchestration.
  • Train and fine-tune custom models, including LLMs and foundation models, to solve unique Upwork-specific challenges.
  • Lead cross-functional collaboration across engineering, product, and research teams to align technical solutions with business impact.
  • Mentor and guide engineers and researchers, helping foster a high-performing, inclusive team culture grounded in AI excellence.
  • Translate experimental ideas into reliable, scalable production systems by applying best practices in AI engineering and deployment.
  • Publish and share innovations where appropriate, contributing to Upwork’s visibility in the broader AI/ML research community.
     

What it takes to catch our eye

  • Strong proficiency in Python and modern ML frameworks such as PyTorch or TensorFlow, with experience developing and deploying AI systems.
  • Deep understanding of core ML concepts, including transformers, generative models, and agent architectures such as MCP or A2A.
  • Experience training large models on GPU clusters and integrating LLMs with external tools via APIs or orchestration frameworks.
  • Comprehensive understanding of foundational deep learning, machine learning concepts, and state-of-the-art GenAI models.
  • Hands-on experience training custom LLMs on GPUs and working with Retrieval-Augmented Generation (RAG) systems.
  • Proven ability to build end-to-end ML pipelines—from data prep and experimentation to production deployment—in a cloud-native environment.
  • A growth mindset, strong communication skills, and the ability to translate complex technical work to stakeholders across functions.
  • Demonstrated experience working in R&D environments; publications in major AI/ML conferences are a plus, but not required.

Come change how the world works.

At Upwork, you’ll shape the future of work for a global, remote-first workforce, creating economic opportunities for professionals worldwide. While we have a physical office in Palo Alto, we currently hire full-time employees in 21 U.S. states, making it easier than ever to join our mission from wherever you call home.

Our culture is built on trust, risk-taking, customer focus, and excellence, all in service of our core mission: to create economic opportunities so people have better lives. We embrace authenticity and inclusion, encouraging everyone to bring their whole selves to work. Personal and professional growth is a priority here, supported through development programs, mentorship, and our Upwork Belonging Communities.

We’re proud to offer benefits that go beyond the basics, including comprehensive medical coverage for you and your family, unlimited PTO, a 401(k) plan with matching, 12 weeks of paid parental leave, and an Employee Stock Purchase Plan. Visit our Life at Upwork page to learn more about our values, working principles, and the overall employee experience.

Ready to help shape the future of work? Check out our Careers page and follow us on LinkedIn, Facebook, Instagram, TikTok, and X to learn more about life at Upwork.

Upwork is an Equal Opportunity Employer committed to recruiting and retaining a diverse and inclusive workforce. We do not discriminate based on race, religion, color, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, or other legally protected characteristics under federal, state, or local law.

Please note that a criminal background check may be required once a conditional job offer is made. Qualified applicants with arrest or conviction records will be considered in accordance with applicable law, including the California Fair Chance Act and local Fair Chance ordinances.

Similar Jobs

PermitFlow - Enterprise Account Executive

PermitFlow

New York, United States (Hybrid)
1 Month ago
PwC - Specialist - Chemicals, Fuels & Resources Advisory

PwC

Mumbai, Maharashtra, India (On-Site)
1 Month ago
AppLovin - Vice President of Marketing

AppLovin

Los Angeles, California, United States (On-Site)
3 Months ago
Mindtickle - Director, Customer Success

Mindtickle

United States (Remote)
3 Months ago
Riot Games - Senior Game Producer, Core Gameplay - Unpublished R&D Product

Riot Games

Los Angeles, California, United States (On-Site)
5 Months ago
Game freak - R&D Programmer: System

Game freak

Chiyoda City, Tokyo, Japan (On-Site)
3 Months ago
Varonis  - DFIR Security Researcher

Varonis

Herzliya, Tel Aviv District, Israel (Hybrid)
7 Months ago
bytedance - Research Scientist Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

bytedance

Seattle, Washington, United States (On-Site)
9 Months ago
bytedance - Research Scientist in Large Language Model, Graduates - Doubao (Seed) - 2024 Start (PhD)

bytedance

Seattle, Washington, United States (On-Site)
9 Months ago
ShyftLabs - AI Engineer

ShyftLabs

Noida, Uttar Pradesh, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Super.com - Senior Full-Stack Software Engineer ( Remote! )

Super.com

Orlando, Florida, United States (Remote)
9 Months ago
appier - Senior Software Engineer, Backend Development (BotBonnie)

appier

Taipei City, Taiwan (On-Site)
3 Months ago
Playstation - Product Manager - Live-Ops

Playstation

Brisbane, Queensland, Australia (On-Site)
1 Month ago
PhonePe - AM, Process Design

PhonePe

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Illumina - Senior Electrical Engineer

Illumina

Singapore (On-Site)
1 Year ago
Figma - Workforce & Vendor Manager, Product Support

Figma

San Francisco, California, United States (Remote)
2 Months ago
Rackspace Technology - Communications and Content Development Manager

Rackspace Technology

San Antonio, Texas, United States (Hybrid)
2 Months ago
Ubisoft - Community Developer

Ubisoft

Montpellier, Occitanie, France (On-Site)
4 Months ago
Apple - Tooling & Automation Engineer, Retail Interactive Technology

Apple

Sunnyvale, California, United States (On-Site)
2 Months ago
Apple - 5G/4G Cellular Layer1 Control Firmware Engineer

Apple

San Diego, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Worldwide

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Research Development Jobs

Apple - Software Engineer - Machine Learning & AI

Apple

Cupertino, California, United States (On-Site)
2 Months ago
bytedance - Tech Lead - Architect / Researcher - DPU

bytedance

Seattle, Washington, United States (On-Site)
6 Months ago
Plaid  - Experienced Machine Learning Engineer

Plaid

San Francisco, California, United States (On-Site)
8 Months ago
Apple - Data ML Program Manager

Apple

Cupertino, California, United States (On-Site)
2 Months ago
Apple - Sr. Applied ML Engineer

Apple

Seattle, Washington, United States (On-Site)
3 Months ago
LMArena - Machine Learning Engineer

LMArena

California, United States (Hybrid)
3 Months ago
Microsoft - Member of Technical Staff, AI Data

Microsoft

London, England, United Kingdom (On-Site)
4 Months ago
Hedra - Research Engineer

Hedra

San Francisco, California, United States (On-Site)
4 Months ago
flying wild hog - Lead User Researcher

flying wild hog

Poland (Remote)
6 Months ago
Apple - AIML - Senior Engineering Program Manager, ML Lifecycle Platform

Apple

Cupertino, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded