Applied AI Engineer
3 days ago
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, , 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 USD - $300,000 USD
-
AI/ML Engineer
1 day ago
Palo Alto, California, United States Nightfall AI Full timeAbout Nightfall:Nightfall is the AI-native, unified data loss prevention and insider risk management platform that protects sensitive data across SaaS apps, GenAI tools, email, endpoint devices, and more. Hundreds of customers, spanning AI innovators to top 10 banks, trust Nightfall to detect and stop data exfiltration at scale. Nightfall enables...
-
Applied AI Engineer, AI Hardware
1 week ago
Palo Alto, California, United States Tesla Full timeWhat To ExpectThe AI Hardware Engineering team at Tesla is transforming EDA tools through AI-driven innovation to supercharge custom silicon for Full Self-Driving and Optimus robotics. We approach EDA as an end-to-end machine intelligence platform: infusing neural networks into synthesis, placement, routing, and verification flows, where every simulation...
-
AI Product Engineer
1 day ago
Palo Alto, California, United States Luma AI Full timeThe Opportunity Luma AI is building the next era of AI with Omni models that can see, hear, and understand the world. As a full-stack company, we control the entire vertical from the GPU cluster to the user interface. We operate with the resources to compete at the forefront while maintaining a lean structure that ensures you will be part of the core...
-
Sr. AI Engineer, Enterprise Applied AI
2 weeks ago
Palo Alto, California, United States Rivian Full time $162,800 - $203,500About RivianRivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what's possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and...
-
Applied AI ML Director
1 day ago
Palo Alto, California, United States JPMorganChase Full timeDescriptionElevate your career as Machine Learning Engineering Lead, where your technical expertise and visionary leadership will shape the future of AI and ML solutions.As an Applied AI ML Director at JPMorgan Chase within the Corporate Sector – AIML Data Platforms, you will lead a specialized technical area, driving impact across teams, technologies, and...
-
AI Deployment Strategist
2 days ago
Palo Alto, California, United States Mistral AI Full timeAbout Mistral At Mistral AI, we believe in the power of AI to simplify tasks, save time, and enhance learning and creativity. Our technology is designed to integrate seamlessly into daily working life. We democratize AI through high-performance, optimized, open-source and cutting-edge models, products and solutions. Our comprehensive AI platform is...
-
AI Physics Engineer
4 days ago
Palo Alto, California, United States DeepSim, Inc. Full timeDeepSim is seeking an AI Physics Engineer to help build the next generation of fast, scalable physics simulation technology. This role is ideal for someone who has expertise at the intersection of modern AI models and physics-based modeling and who is excited to transform how engineers simulate complex physical systems.ResponsibilitiesDesign and implement...
-
Staff Security Engineer
3 days ago
Palo Alto, California, United States Fiddler AI Full timeOur PurposeAt Fiddler, we understand the implications of AI and the impact that it has on human lives. Our company was born with the mission of building trust into AI. The rise of Generative AI and Agents has unlocked generalized intelligence but also widened the risk aperture and made it harder to ensure that AI applications are working well. Fiddler...
-
New Grad Engineer
1 week ago
Palo Alto, California, United States Nightfall AI Full timeAbout Nightfall:Nightfall is the AI-native, unified data loss prevention and insider risk management platform that protects sensitive data across SaaS apps, GenAI tools, email, endpoint devices, and more. Hundreds of customers, spanning AI innovators to top 10 banks, trust Nightfall to detect and stop data exfiltration at scale. Nightfall enables...
-
Product Engineer
2 weeks ago
Palo Alto, California, United States Luma AI Full time $105,000 - $175,000 per yearThe Opportunity Luma AI is defining the future of autonomous creativity. We combine the resources of a capitalization-heavy AI lab with the speed of an elite product team. This environment guarantees that you will sit at the intersection of Product, Design, and Research, translating raw model capabilities into magical user experiences that define a new...