AI Product Engineer: End-to-End Agent & UI Builder

tryalma; Palo Alto (CA); On-site; USD 120,000 - 210,000; Full time

Benefits offered by this job

Hyper-competitive base salary

Comprehensive health coverage

PTO + federal holidays

Wellness allowance

Job summary

Alma, an AI-native law firm in Palo Alto, is building a world-class immigration platform that blends a legal team with cutting-edge AI. We seek a Product Engineer (AI) who owns end-to-end features, designs agentic pipelines, and builds the React frontend the user interacts with.

You’ll work closely with the founding team and early customers to ship production-ready AI tooling, iterate rapidly, and push the boundaries of what an AI-powered legal product can do.

Qualifications

• 2+ years shipping production AI/LLM features.
• Strong knowledge of RAG architectures, prompt engineering, agentic frameworks, and LLM orchestration.
• Experience designing evaluation and quality systems for AI outputs.
• Proficient in Python, FastAPI, PostgreSQL, containers.
• Comfortable with TypeScript, React, and modern frontend tooling.
• Daily, hands-on use of modern coding harnesses (Claude Code, Codex, Cursor).
• Located in the San Francisco Bay Area or willing to relocate.
• BS/MS in Computer Science, Engineering, or related technical field.

Responsibilities

• Build world-class AI-powered product with founding team.
• Own AI-powered features end-to-end: design agentic pipelines, LLM integrations, backend services, and frontend.
• Design, build, and iterate on RAG pipelines, agentic workflows, prompt chains, and evaluation frameworks.
• Develop and maintain evaluation and observability systems to measure AI outputs and catch regressions.
• Partner with designers and product to translate user needs into shipped AI-powered experiences.
• Leverage AI tooling to move fast while maintaining quality across code, reviews, and tests.
• Engage in the entire application lifecycle with high coding and debugging standards.

Skills

Production AI/LLM features

RAG architectures

Prompt engineering

LLM orchestration

Python

FastAPI

PostgreSQL

Containerization

TypeScript

React

MCP servers

Education

BS/MS in Computer Science or related field

Tools

MCP servers

Job description

Alma, an AI-native law firm in Palo Alto, is building a world-class immigration platform that blends a legal team with cutting-edge AI. We seek a Product Engineer (AI) who owns end-to-end features, designs agentic pipelines, and builds the React frontend the user interacts with.

You’ll work closely with the founding team and early customers to ship production-ready AI tooling, iterate rapidly, and push the boundaries of what an AI-powered legal product can do.