Staff AI Engineer

This is Engineering at Lattice

Lattice's Engineering team is continuously working to better both our product and our craft. We use a modern, cutting-edge tech stack aimed at developer productivity and delight. We strive for maintainable, robust, and performant code. We're highly collaborative and continuously iterative and work closely with designers and product managers. We prioritize not only great technical architecture but also an amazing product experience.

What You Will Do

You will join Lattice’s AI Engineering team, where you will lead and shape Lattice's artificial intelligence strategy while building software to help the humans powering organizations thrive, with a particular focus on large language models (LLMs) and emerging machine learning technologies.

• You will design and build end-to-end architectures for AI/ML solutions, including data pipelines, model development, and production deployment.
• You will develop, fine-tune, and optimize LLM-based solutions for Lattice’s talent platform and product suite.
• You will drive architectural decisions, balancing trade-offs between scalability, latency, accuracy, and cost.
• You will implement AIOps practices, including CI/CD pipelines, feature flagging, model monitoring, and performance optimization.
• You will work at a strategic level to influence technical decisions by collaborating cross-functionally with product managers, data scientists, and engineers to deliver AI-driven features aligned with business goals.
• You will mentor and guide engineers of various experiences in AI engineering, establishing best practices for software and AI development, standardizing and evangelizing AI engineering in our team.
• You'll lead technical evaluation of AI/ML vendors and platforms, making strategic recommendations on build vs. buy decisions for our AI infrastructure, from LLM providers to security and monitoring solutions.

What You Will Bring to the Table

• You have 8+ years of professional experience writing and maintaining production-level applications, with at least 4 years in designing and implementing scalable AI/ML systems in production.
• You have deep expertise in LLM concepts including prompt engineering, AI feature evaluation metrics, model fine-tuning, knowledge of vector DBs, etc. 
• You have experience with advanced LLM techniques, at a minimum with Retrieval-Augmented Generation (RAG), and understanding or exposure to RLHF, or LoRA.
• You have experience or knowledge of ML traditional techniques (supervised / unsupervised systems or neural networks) and be able to pin point when a solution requires an LLM solution or an ML solution
• You have strong programming skills in Python, with hands-on experience in frameworks like PyTorch, TensorFlow, Hugging Face Transformers, or LangChain.
• You have expertise in MLOps: deployment (Docker, Kubernetes, Terraform and/or CDK), model monitoring (MLflow, DataDog, etc), and CI/CD workflows.
• You are proficient in the AWS ecosystem (Lambdas, SQS, SNS, etc) and data engineering tools for large-scale datasets.
• You have excellent problem-solving, system design, and technology decision-making skills.
• You have strong communication and collaboration skills to lead projects and mentor junior team members, and to also collaborate with product and executives.

Nice to Have

• Familiarity with hybrid solutions combining traditional ML and LLMs. 
• Experience with agentic AI or building multi agent systems
• Experience with cloud-based ML services (e.g., AWS SageMaker, Google Cloud AI, Azure Machine Learning)
• Experience with TypeScript / Javascript as it is our current programming language (Our current stack includes NodeJS, GraphQL, and PostgreSQL. We rely on AWS to host our infrastructure)
• Professional experience with multiple cloud platforms (AWS + others such as Azure or GCP) 
• Track record of publishing or presenting on artificial intelligence and machine learning topics in professional forums

The estimated annual cash salary for this role is $187,500 - $234,500. This position is also eligible for incentive stock options, subject to the terms of Lattice’s applicable plans.

Benefits: The Company offers the following benefits for this position, subject to applicable eligibility requirements: Medical insurance; Dental insurance; Vision insurance; Life, AD&D, and Disability Insurance; Emergency Weather Support; Wellness Apps; Paid Parental Leave, Paid Time off inclusive of holidays and sick time; Commuter & Parking Accounts; Lunches in the Office; Workplace Amenities Stipend, Internet and Phone Stipend; One time WFH Office Set-Up Stipend; 401(k) retirement plan; Financial Planning; Learning & Development Budget; Sabbatical Program; and Invest in Your People Fund

*Note on Pay Transparency:

Lattice provides an estimate of the compensation for roles that may be hired as required by state regulations. Compensation may vary based on (a) location, as Lattice factors in specific location when benchmarking compensation for most roles; (b) individual candidate skills and qualifications; and (c) individual candidate experience.

Additionally, Lattice leverages current market data to determine compensation, so posted compensation figures are subject to change as new market data becomes available. The salary, other compensation, and benefits information is accurate as of the date of this posting. Lattice reserves the right to modify this information at any time, subject to applicable law.