AI Product Engineer, Tech Lead

Palo Alto, California, United States

Apply now Apply later

About Augment Code

The best software comes from Augmenting developers, not replacing them. We’re bringing joy back to software engineering and keeping developers in flow by building the best Developer AI, with deep contextual awareness of a company’s codebase.

About the role

The world of software development is on the brink of transformation — and Augment is leading the charge. We’re building the next generation of AI-powered tools that help engineering teams move faster, stay in flow longer, and deliver more impactful outcomes.

As an AI Product Engineer at Augment, you’ll be at the heart of this transformation.

In this role you will:

  • Lead the ideation, design, development, and launch of AI-powered experiences that boost developer productivity.
  • Integrate cutting-edge models into seamless, delightful product flows.
  • Use an empirical mindset to experiment, measure, and optimize for maximum user value.
  • Stay ahead of the curve by embracing the rapid pace of AI advancements.
  • Mentor teammates on product thinking and contribute to a culture of continuous learning.
  • Partner with design and research to shape user insights into product enhancements.

You have:

  • 6+ years of software engineering and product experience
  • A passion for building elegant tools that simplify complex workflows for developers.
  • Experience creating clarity in ambiguous problem spaces and delivering polished products.
  • Strong coding skills in one or more of the following: Node.js, JavaScript, TypeScript, Svelte, React, Kotlin, Java — or similar.
  • Experience with prompting LLMs and tuning responses to create moments of user delight.
  • A proactive approach to solving problems and collaborating across functions.

While not required, it’s an added plus if you also have:

  • Shipped AI-native products built on top of foundation models (e.g., OpenAI, Claude, Gemini).
  • Experience with making AI outputs more reliable and building user trust in AI-powered features.
  • Familiarity with cloud platforms (especially GCP).

Employee Benefits:

  • Flexible work hours
  • Competitive salary & Equity
  • Tools Stipend
  • Health, Dental, Vision and Life Insurance
  • Short Term and Long Term Disability
  • Unlimited Paid Time Off + Holidays. We focus on trust and ownership, not time in the chair
  • Numerous company social events

We will do everything we can within reason to make sure that your interview takes place in an environment that fairly and accurately assesses your skills. If you need assistance or accommodation, please contact your recruiter.

Augment Code is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon 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, or other applicable legally protected characteristics.

By applying for this job, the candidate acknowledges and agrees that any personal data contained in their application or supporting materials will be processed in accordance with Augment Code's Applicant Privacy Policy.

Pay Transparency Notice: The actual salary within the stated range will be based on a combination of factors such as an individual's skills, experience level, educational background, and other relevant job-related considerations

Pay Range$225,000—$300,000 USD
Apply now Apply later

Tags: Claude Engineering GCP Gemini Java JavaScript LLMs Node.js OpenAI Privacy Prompt engineering React Research TypeScript

Perks/benefits: Career development Competitive pay Equity / stock options Flex hours Flex vacation Health care Insurance Team events Unlimited paid time off

Region: North America
Country: United States

More jobs like this