Lead Software Engineer - AI Data Systems

19 Hours ago • All levels • $151,500 PA - $215,750 PA

Job Summary

Job Description

The Lead Software Engineer (AI Data Systems) will be responsible for building the infrastructure that powers the future of intelligent, agent-driven systems at Upwork. This includes collecting high-quality training data, building scalable featurization pipelines, and delivering performant systems to support model training and inference at scale. The role involves collaborating with ML engineers and researchers, and mentoring engineers. The engineer will design and implement systems for data collection, build pipelines, support model deployment, and drive cross-functional alignment. This role requires strong experience in software engineering with deep expertise in building data collection and transformation pipelines.
Must have:
  • Build data collection, transformation and featurization pipelines.
  • Proficiency in Python and data-centric frameworks.
  • Experience deploying and scaling AI systems in cloud environments.
  • Experience with ML model development workflows.
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 Software Engineer (AI Data Systems) to join our growing AI team and help build the critical infrastructure that powers the future of intelligent, agent-driven systems at Upwork. You’ll be responsible for collecting high-quality training data, building scalable featurization pipelines, and delivering performant systems to support model training and inference at scale.

This role is built for Explorers, engineers who thrive in ambiguity, move fast with instinct, and are energized by greenfield challenges. Instead of narrowly scoped specialists, we’re hiring T-shaped engineers: individuals with a breadth of skills across the stack and deep expertise in at least one technical area. You’ll be one of our foundational hires, pushing past the known, building fast in Python, and helping us shape our technology and our team. Experience in startups, AI/ML research environments, or similarly dynamic settings is essential, as we’re solving problems that demand speed, curiosity, and adaptability.

If you’re passionate about AI infrastructure, thrive in early-stage environments, and want to help us set the standard for how agentic systems are built and deployed at scale, this is your opportunity.

Responsibilities

  • Design and implement systems to collect and curate high-quality training datasets for supervised, unsupervised, and reinforcement learning use cases.
  • Build scalable featurization and preprocessing pipelines to transform raw data into structured inputs for AI/ML model development.
  • Partner with ML engineers and researchers to define data requirements and production workflows that support LLM-based agents and autonomous AI systems.
  • Lead the development of infrastructure that enables experimentation, evaluation, and deployment of machine learning models in production environments.
  • Support orchestration and real-time inference pipelines using Python and modern cloud-native tools, ensuring low-latency and high availability.
  • Mentor engineers and foster a high-performance, collaborative engineering culture grounded in technical excellence and curiosity.
  • Drive cross-functional alignment with product, infrastructure, and research stakeholders, ensuring clarity on progress, goals, and architecture.

What it takes to catch our eye

  • Strong software engineering background with deep experience in building data collection, transformation, and featurization pipelines at scale.
  • Proficiency in Python, including async programming and concurrency tools, as well as data-centric frameworks such as Pandas, Spark, or Apache Beam.
  • Familiarity with ML model development workflows and infrastructure, including dataset versioning, experiment tracking, and model evaluation.
  • Experience deploying and scaling AI systems in cloud environments such as AWS, GCP, or Azure.
  • Proven success operating in highly ambiguous environments such as research labs, startups, or fast-paced product teams.
  • A track record of working with or alongside high-caliber peers in top engineering teams, research groups, or startup ecosystems.
  • Growth mindset, strong communication skills, and a commitment to inclusive collaboration and continuous learning.

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

Meta - Research Scientist Intern, Language and Multimodal Research for MetaAI (PhD)

Meta

New York, New York, United States (On-Site)
5 Months ago
Scale AI - Senior Fullstack Software Engineer, GenAI Tasking Experience

Scale AI

San Francisco, California, United States (Hybrid)
1 Day ago
Scale AI - Senior Fullstack Software Engineer, GenAI Model Quality

Scale AI

San Francisco, California, United States (Hybrid)
1 Day ago
Lucid Reality Labs - Machine Learning Engineer

Lucid Reality Labs

Poland (Remote)
3 Months ago
Scale AI - Senior Software Engineer, Pay & Incentives

Scale AI

San Francisco, California, United States (Hybrid)
1 Day ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Machine Learning Engineer Intern (Global E-commerce Risk Control) - 2025 Summer (PhD)

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Google - Software Engineer III, AI/ML

Google

Hyderabad, Telangana, India (On-Site)
2 Weeks ago
Krafton  - Head of Deep Learning PM & Ops Dept.

Krafton

Seoul, South Korea (On-Site)
1 Month ago
NVIDIA - Senior System Software Engineer, Robotics Simulation

NVIDIA

Santa Clara, California, United States (Hybrid)
1 Month ago
ByteDance - Research Scientist for Generative AI, Multimodal and LLM

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Scale AI - Engineering Manager, Connectivity

Scale AI

San Francisco, California, United States (Hybrid)
1 Day ago
P99 soft - Data Engineer

P99 soft

Hyderabad, Telangana, India (On-Site)
1 Day ago
Zazz - Artificial Intelligence Engineer

Zazz

(Remote)
2 Months ago
Netflix - Machine Learning Intern - Spring or Summer 2025

Netflix

Los Gatos, California, United States (On-Site)
6 Months ago
ByteDance - Software Engineer, ML System Architecture

ByteDance

San Jose, California, United States (On-Site)
6 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!

Similar Category Jobs

Looks like we're out of matches

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

About The Company

Upwork is the world’s work marketplace that connects businesses with independent talent from across the globe. We serve everyone from one-person startups to large, Fortune 100 enterprises with a powerful, trust-driven platform that enables companies and talent to work together in new ways that unlock their potential. Our talent community on Upwork encompasses more than 10,000 skills in categories including website & app development, creative & design, customer support, finance & accounting, consulting, and operations.

View All Jobs

Get notified when new jobs are added by Upwork

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug