Lead AI Software Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Klaviyo · 3 months ago

Lead AI Software Engineer

Klaviyo is a company that empowers creators to own their destiny by making first-party data accessible and actionable. They are seeking a Lead AI Software Engineer to set technical direction and drive the design of scalable backend systems and AI-powered user experiences, while mentoring engineers and shaping architectural decisions.

AdvertisingAnalyticsE-CommerceMarketing AutomationSoftware
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Lead the design and development of backend systems that enable scalable AI solutions for 167K+ customers
Architect robust, reliable, and scalable data pipelines that train and serve AI/ML models at scale
Drive the integration of LLMs into production systems, powering AI agents that create and execute real-time customer experiences
Establish best practices for model evaluation, monitoring, and performance across generative and agentic AI use cases
Mentor and guide engineers, fostering a culture of technical excellence, ownership, and customer-centric product thinking
Partner with cross-functional leaders to influence AI product strategy and deliver high-impact solutions

Qualification

AI/ML experienceLarge Language ModelsBackend systems designDistributed architecturesPython proficiencyBig data toolsAPI architectureCloud-native solutionsMentoring engineersCuriosityTechnical leadershipAdaptability

Required

7–10 years of professional software engineering experience, with a strong focus on backend systems and distributed architectures
Recent, hands-on experience with AI/ML and LLMs, including fine-tuning, evaluation, and production deployment
Proven ability to design and scale distributed systems that support AI-driven applications and agent capabilities
Proficient in Python and modern backend frameworks (FastAPI, Django preferred)
Skilled with big data tools (Apache Spark, Hadoop) and distributed task queues (Kafka, Celery, SQS, RabbitMQ, Redis)
Experienced in architecting APIs and cloud-native solutions (AWS, Kubernetes, CI/CD pipelines)
Demonstrated technical leadership — mentoring engineers, shaping architecture, and raising the bar for engineering practices
Thrives in a fast-paced, startup-like environment with high autonomy and ownership
Curious, adaptable, and motivated to stay on the cutting edge of AI/ML research and apply it to real-world business challenges

Preferred

Experience training and deploying reinforcement learning systems
Prior experience leading cross-functional initiatives involving data science, ML engineering, and product

Benefits

Participation in the company’s annual cash bonus plan
Variable compensation (OTE) for sales and customer success roles
Equity
Sign-on payments
A comprehensive range of health, welfare, and wellbeing benefits based on eligibility

Company

Klaviyo is an automation and email platform designed to help grow businesses.

H1B Sponsorship

Klaviyo has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (47)
2024 (29)
2023 (24)
2022 (27)
2021 (21)
2020 (8)

Funding

Current Stage
Public Company
Total Funding
$1.35B
Key Investors
ShopifySands Capital VenturesAccel
2025-08-13Post Ipo Secondary· $195.06M
2025-05-14Post Ipo Secondary· $372.95M
2023-09-20IPO

Leadership Team

leader-logo
Andrew Bialecki
CEO
linkedin
leader-logo
Ed Hallen
Co-Founder, Chief Strategy Officer, Board Member
linkedin
Company data provided by crunchbase