AI Engineer
Listed on 2026-02-24
-
Software Development
AI Engineer, Machine Learning/ ML Engineer
About the Role
We are seeking an exceptional AI Engineer to join our growing team and take a leading role in building and evolving our internal AI-powered agent platform used across the organization.
In this role, you will design, build, and improve AI-powered applications that integrate large language models (LLMs) with internal tools, SQL databases, advertising APIs (Google Ads, Meta), and stateful code execution environments. You will play a critical role in improving reliability, context orchestration, evaluation maturity, and expanding Rai into document and workflow automation use cases.
Your work will directly impact 200–500 employees and potentially clients, making AI a core productivity multiplier across the organization.
Responsibilities- LLM Application Development Design, build, and maintain AI-powered applications leveraging LLM APIs, tool calling, and structured outputs.
- Agent & Tool Orchestration
Develop and improve agent architectures that integrate tools such as SQL execution, Python code execution, and marketing platform APIs through MCP-style integrations. - Context & Memory Management Design and refine multi-turn context handling, session memory strategies, and prompt structures to improve response reliability and reduce hallucinations.
- Reliability & Guardrails
Implement safeguards for tool execution, SQL queries, and code execution. Improve retry logic, structured validation, and hallucination mitigation strategies. - Evaluation & Quality Assurance
Contribute to building structured evaluation frameworks for LLM-powered systems, including multi-turn testing, regression testing, and task-based success metrics. - Product Expansion
Support expansion of Rai into document generation, tables, slide decks, and potential Google Workspace integrations. - Collaboration
Work closely with Machine Learning Engineers, Product Managers, and Full-Stack Engineers to ensure AI systems are robust, maintainable, and aligned with business goals. - Continuous Innovation
Stay current with advances in LLM systems, agent architectures, and applied AI tooling to continuously improve Rai’s capabilities.
- Strong programming skills in Python with experience building production systems.
- Experience building applications powered by LLM APIs (OpenAI, Anthropic, Vertex AI, etc.).
- Deep understanding of:
- Context windows and token limits
- Multi-turn conversation behavior
- Tool/function calling
- Structured outputs (JSON schema)
- Prompt design and failure modes
- Experience integrating external APIs into production systems.
- Proficiency in SQL.
- Experience working with cloud platforms (GCP, AWS, or Azure).
- Strong problem-solving skills and ability to debug complex system behavior.
- Excellent communication and cross-functional collaboration skills.
- Experience designing evaluation frameworks for LLM-powered systems.
- Understanding of embeddings, vector search, and RAG architectures.
- Experience building agent or multi-agent workflows.
- Experience with stateful execution environments or sandboxed code execution.
- Experience with document automation (Google Docs, Slides, PDFs).
- Experience building internal developer tools or productivity platforms.
- Type Script.
- Python
- SQL
- LLMs / AI APIs
- Git / Git Hub
- Data Warehouses (Big Query, Snowflake, Redshift)
- Data Transformation (dbt)
- Semantic Layers (Cube, Looker, dbt Metrics)
- Type Script
- Excel and Typing Test
- Unlimited vacation policy
- Monthly Phone Stipend
- Comprehensive Medical, Dental, and Vision insurance options
- 401(K) plan with matching
- Dog friendly office
- Hybrid work opportunity
- Professional Development Program
- Bonus Perk - Seamless allowance
Base salary based on education, experience, and skills level ($145,057-$296,054)
$145,057-$175,256
$175,256-$205,456
Level 3Posesses notable capabilities.
$205,456-$235,655
Level 4Posesses strong capabilities.
$235,655-$265,855
Level 5Posesses advanced capabilities.
$265,855-$296,054
About WITHIN & Brkfst.ioWITHIN is the world's first Performance Branding company, partnering with some of the biggest brands in the world…
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).