FlexHired LogoFlexHired
Logo of Upwork

Upwork

Lead Machine Learning Engineer - AI Agents

Job Summary

The role involves designing and implementing infrastructure and models to support agent-based large language model (LLM) systems, with a focus on delivering advanced machine learning solutions that align with business goals. Candidates should have strong experience with Python, ML frameworks like PyTorch or TensorFlow, and training large models on GPU clusters. The position requires collaboration across engineering, product, and research teams, along with mentoring and fostering a high-performance, inclusive team culture. Emphasis is placed on translating experimental ideas into scalable production systems and contributing to AI research through publications.

Required Skills

Python
Communication
PyTorch
TensorFlow
Research & Development
ML Frameworks
ML Pipelines
Transformers
Multi-agent Systems
Retrieval-Augmented Generation
AI Deployment
Agent Architectures
Generative Models
GPU training

Benefits

Paid Parental Leave
Employee Stock Purchase Plan
Medical Coverage
Unlimited PTO
401(k) plan with matching

Job Description

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.

The annual base salary range for this position is displayed below. The range displayed reflects the minimum and maximum salary for this position, and individual base pay will depend on your skills, qualifications, experience, and location. Additionally, this position is eligible for the annual bonus plan or sales incentive plan and eligibility to participate in our long term equity incentive program.

Annual Base Compensation
$175,500$277,500 USD

To learn more about how Upwork processes and protects your personal information as part of the application process, please review our Global Job Applicant Privacy Notice

Interested in this job?

Application deadline: Open until filled

Logo of Upwork

Upwork

A freelancing platform connecting businesses with independent professionals for remote work across various industries.

See more jobs
Date PostedMay 30th, 2025
Job TypeFull Time
LocationRemote
Salary$175,500 - $277,500
Exciting fully remote opportunity for a Lead Machine Learning Engineer - AI Agents at Upwork. Offering $175,500 - $277,500 (full time). Explore more remote jobs on FlexHired!

Safe Remote Job Search Tips

Verify Employer Thoroughly

Research the company's identity thoroughly before applying. Check for a professional website with contacts, active social media, and LinkedIn profiles. Verify details across platforms and look for reviews on Glassdoor or Trustpilot to confirm legitimacy.

Never Pay to Get a Job

Legitimate employers never require payment for applications, training, background checks, or equipment. Always reject upfront payment requests or demands for bank details, even if they claim it's for purchasing necessary work gear on your behalf.

Safeguard Your Personal Information

Protect sensitive data like SSN, bank details, or ID copies. Share this only after accepting a formal, written job offer. Ensure it's submitted via a secure company system or portal, never through insecure channels like standard email attachments.

Scrutinize Communication & Interviews

Watch for communication red flags: poor grammar, generic emails (@gmail), vague details, or undue pressure. Be highly suspicious of interviews held only via text or chat apps; legitimate companies typically use video or phone calls.

Beware of Unrealistic Offers

If an offer's salary or benefits seem unrealistically high for the work involved, be cautious. Research standard pay for similar roles. Offers that appear 'too good to be true' are often scams designed to lure you into providing information or payment.

Insist on a Formal Contract

Always secure and review a formal, written job offer or employment contract before starting work or sharing final personal details. Ensure it clearly defines your role, compensation, key terms, and conditions to avoid misunderstandings or scams.

Related Jobs

Full Time
$190,800 - $267,100
Remote - United States
Full Time
$202,500 - $253,000
Remote, USA
Full Time
$190,000 - $240,000
Remote US
Full Time
$230,000 - $322,000
Remote - United States
Full Time
$230,000 - $322,000
Remote - United States

Subscribe Newsletter

Never miss a remote job opportunity. Subscribe to our newsletter today and receive exclusive job alerts, career advice, and industry insights delivered straight to your inbox.