Staff AI Engineer

The Role

As a Staff AI Engineer, you will be a technical multiplier and a strategic partner in our journey to leveraging AI to make the world financially strong. You will lead the design and implementation of production AI systems while ensuring that AI adoption is seamless and safe for the rest of the engineering organization.

We are looking for an AI Engineer with broad technical mastery who operates with high integrity, exhibits empathy for their peers and customers, and has a growth and teaching mindset. You know that your success is measured by quality and velocity metrics as well as how effectively your colleagues can leverage the tools and platforms you build. We’re looking for world-class builders and collaborators, not research scientists. We understand that an AI generalist won’t be an expert in every one of the bullet points below. If you are missing experience in some of these areas, we still want to speak with you.

Key Responsibilities

• Architect Production RAG Systems: Design and scale Retrieval-Augmented Generation pipelines, optimizing for precision and recall while managing the complexities of financial data structures.

• LLM Orchestration & Governance: Implement LLM gateways to handle provider failover, load balancing, and prompt caching. You will be part of the team controlling our cost, latency, and availability metrics. Advise when a use case is best served by AI, and when to avoid it. Governance should include processes to validate use cases and operational monitoring for policy compliance.

• Model Optimization & SFT: Identify opportunities to leverage Supervised Fine-Tuning (SFT) of existing models. You know when to prune a model for efficiency and how to curate high-quality synthetic data for training.

• Context Engineering: Master dynamic context window management to ensure models are "well-informed" without exceeding token budgets or inducing "lost in the middle" phenomena.

• Agentic Workflows & Tool Use: Design frameworks that allow LLMs to safely interact with internal financial APIs, moving beyond simple chat to autonomous task execution within strict guardrails.

• Technical training and leadership: Provide guidance and training for product engineering teams.

• Security: Design and implement systems that enforce strict access and authorization to data, validation of outputs, and enable integrity and non-repudiation, even with LLMs in the stack.

• The "Eval" Moat: Build automated evaluation frameworks (LLM-as-a-judge) to quantify model performance, ensuring that "hallucinations" are caught long before they reach a customer.

Culture & Leadership

• Mindset: You possess the judgment to bypass the hype cycle and cut through analysis paralysis. You take calculated risks with a clear plan for mitigation, knowing exactly when to innovate and when to rely on battle-tested, proven patterns. You communicate and collaborate through every step of your process.

• Empathy: You treat your fellow technologists and product managers as your primary customers. You build abstractions that make it easy for others to do the right thing and hard to do the wrong thing.

• Integrity: You possess the integrity to view security and privacy controls not as hurdles, but as the infrastructure that enables us to move quickly and safely. You have the courage to flag risks, own your mistakes, and tell the truth even when it’s uncomfortable or inconvenient.

Must haves:

• Demonstrated experience designing systems that capture production feedback to create “ground truth” datasets, turning our operational exhaust into a competitive advantage.

• Expertise with BigQuery and VertexAI

• Production experience with Python, langchain, and langgraph

• Experience tuning and monitoring models in production environments

• Experience with PCI or other highly regulated environments

Nice to haves:

• AWS Bedrock and SageMaker

• Data engineering experience

At MX, we are a high-performance organization that thrives on trust and results. This role is based in Lehi, Utah, with flexibility for both in-office and remote work. We believe in empowering our team members to deliver exceptional outcomes while taking advantage of our incredible office space when it best supports their work. Our Utah office features onsite perks such as company-paid meals, massage therapists, a sports simulator, gym, mother’s lounge, and meditation room and meaningful interactions with amazing people. We encourage team members to come together in the office to collaborate, kick off key projects, or strategize cross-functionally, fostering connection and innovation.

MX is proudly committed to recruiting and retaining a diverse and inclusive workforce. As an Equal Opportunity Employer, we never discriminate based on race, religion, color, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, age, military or veteran status, status as an individual with a disability, or other applicable legally protected characteristics. We particularly welcome applications from veterans and military spouses. All your information will be kept confidential according to EEO guidelines. You may request reasonable accommodations by sending an email to hr@mx.com.