Applied AI Engineer

Posted 9 Days Ago
Easy Apply
Palo Alto, CA
In-Office
225K-300K Annually
Mid level
Artificial Intelligence • Software
The Role
As an Applied AI Engineer, you will implement AI capabilities into software products, collaborating with teams to enhance developer experience through LLM features and integrations.
Summary Generated by Built In
About Augment Code

Augment Code is the only AI coding assistant built for professional software engineers working in large, production‑grade codebases. Our Context  Engine understands your entire repo, enabling developers to stay in flow while writing, reviewing, and understanding code. Backed by top‑tier investors and trusted by engineering teams at leading tech companies, Augment Code is redefining how modern software is built.

About the role

We’re looking for an Applied AI Engineer to help bring cutting-edge AI capabilities into the hands of developers. This is a hands-on engineering role at the intersection of product, AI, and systems — focused on implementing and integrating LLM-powered features that enhance developer experience and productivity.

You’ll collaborate with engineers, researchers, and designers to translate product needs into intuitive, reliable AI-powered experiences. You’ll also play a key role in shaping how AI is used across our stack — from prompt design to system integration — while staying up to date on emerging AI capabilities.

Examples of what you could do:

  • Implement and integrate AI functionality into key product features
  • Craft and iterate on prompts to improve LLM reliability and usefulness
  • Build AI-powered flows that feel intuitive and responsive to developers
  • Evaluate and test AI outputs to ensure performance and accuracy
  • Work alongside engineers to deliver robust, production-grade code
  • Stay current with LLM tools, APIs, and best practices
You will…
  • Deliver reliable, high-quality AI-powered product experiences
  • Translate product needs into technical AI implementations
  • Tune and test prompts for real-world use cases and developer workflows
  • Collaborate closely with engineers and researchers
  • Contribute across frontend, backend, and integration layers
Qualifications:
  • Strong coding skills in one or more of: Python, Go, Node.js, JavaScript, TypeScript, React, or Java
  • Experience with API integrations and service-oriented architectures
  • Familiarity with prompt engineering for LLMs (e.g. OpenAI, Claude, Gemini)
  • Ability to evaluate and optimize AI outputs for reliability and quality
  • Strong problem-solving instincts and attention to detail
  • Collaborative mindset and eagerness to learn
Bonus Points:
  • Experience building product features that incorporate LLMs
  • Understanding of best practices for AI reliability and safety
  • Background in frontend development or UX-oriented implementation
  • Familiarity with cloud platforms (especially GCP)
  • Basic understanding of LLM behavior, strengths, and limitations
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 base 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. 

Annual Base Salary Range
$225,000$300,000 USD

Top Skills

Go
Java
JavaScript
Node.js
Python
React
Typescript
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Palo Alto, California
89 Employees
Year Founded: 2022

What We Do

Augment puts your team’s collective knowledge—codebase, documentation, and dependencies—at your fingertips via chat, code completions, and suggested edits.

Get up to speed, stay in the flow and get more done. Lightning fast and highly secure, Augment works in your favorite IDEs and Slack.

We proudly augment developers at Webflow, Kong, Pigment, and more.

We are alumni of great AI and cloud companies, including Google, Meta, NVIDIA, Snowflake, and Databricks.

If, like us, you believe in augmenting and not replacing software developers, join us on our mission to improve software development at scale using AI.

Similar Jobs

Block Logo Block

Staff Engineer

Blockchain • eCommerce • Fintech • Payments • Software • Financial Services • Cryptocurrency
In-Office
8 Locations
12000 Employees
168K-297K Annually

Granica Logo Granica

Research Engineer - Applied AI

Artificial Intelligence • Big Data • Cloud • Machine Learning • Software • Business Intelligence • Data Privacy
In-Office
Mountain View, CA, USA
37 Employees
160K-250K Annually

Block Logo Block

Senior Machine Learning Engineer

Blockchain • eCommerce • Fintech • Payments • Software • Financial Services • Cryptocurrency
In-Office
8 Locations
12000 Employees
195K-343K Annually

NVIDIA Logo NVIDIA

Senior Timing CAD Engineer, Applied AI

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
In-Office
2 Locations
21960 Employees
136K-213K Annually

Similar Companies Hiring

PRIMA Thumbnail
Travel • Software • Marketing Tech • Hospitality • eCommerce
US
15 Employees
Scotch Thumbnail
Software • Retail • Payments • Fintech • eCommerce • Artificial Intelligence • Analytics
US
25 Employees
Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account