Klaviyo · 1 day ago
Lead AI Software Engineer
United States
Full-time
Remote
Senior Level, Lead/Staff
$188K/yr - $282K/yr
7+ years expKlaviyo is a company that empowers creators by making first-party data accessible and actionable. They are seeking a Lead AI Software Engineer to design, build, and scale backend and machine learning systems that support their AI strategy, collaborating with various teams to deliver intelligent solutions and mentor other engineers.
AdvertisingAnalyticsE-CommerceMarketing AutomationSoftware
Responsibilities
Lead the design and development of backend and ML systems that scale to serve 167K+ customers
Build robust, reliable, and efficient data processing and model-serving pipelines that power production ML models
Develop and optimize predictive systems (e.g., conversion likelihood, engagement ranking, personalization, anomaly detection)
Architect systems and services that enable our AI agents to learn and improve autonomously from reward or outcome signals
Collaborate cross-functionally to translate business challenges into scalable, measurable ML solutions
Collaborate with software engineers and product managers to turn predictive models into intuitive product features for our customers
Mentor engineers and help evolve Klaviyo’s ML engineering best practices, tooling, and architecture
Foster a culture of ownership, experimentation, and data-driven innovation across teams
Qualification
Required
7–10 years of experience in machine learning engineering or applied software engineering, with a focus on classical ML and predictive modeling
Deep understanding of supervised learning methods such as random forests, XGBoost, logistic regression, and ensemble techniques
Proven track record of building and scaling production-grade ML systems end to end (data pipelines, training, deployment, and monitoring)
Experienced backend engineer comfortable with distributed systems and high-scale architectures
Proficient in Python and modern frameworks (FastAPI, Django preferred)
Skilled with big data and streaming technologies (Apache Spark, Kafka, Hadoop)
Hands-on with asynchronous processing and distributed task queues (Celery, SQS, RabbitMQ, Redis)
Strong understanding of databases and ORMs (SQLAlchemy, Alembic) and cloud-native architectures (AWS, Kubernetes, CI/CD)
Excellent communicator who can set technical direction, influence product strategy, and mentor junior engineers
Curious, pragmatic, and passionate about building AI systems that drive measurable business impact
Preferred
Experience training and deploying ML models in production environments that generated measurable impact
Familiarity with reinforcement learning or agentic AI systems that adapt based on outcome or reward signals
Benefits
Participation in the company’s annual cash bonus plan
Variable compensation (OTE) for sales and customer success roles
Equity
Sign-on payments
Comprehensive range of health, welfare, and wellbeing benefits based on eligibility
Company
Klaviyo
Klaviyo is an automation and email platform designed to help grow businesses.
1001-5000 employees
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 CompanyTotal Funding
$1.35BKey Investors
ShopifySands Capital VenturesAccel
2025-08-13Post Ipo Secondary· $195.06M
2025-05-14Post Ipo Secondary· $372.95M
2023-09-20IPO
Leadership Team
Andrew Bialecki
CEO
Ed Hallen
Co-Founder, Chief Strategy Officer, Board Member
Recent News
Investing.com
2026-01-03
2025-12-17
2025-12-16
Company data provided by crunchbase