Senior AI Engineer

Klaviyo is a company that empowers creators to own their own destiny by making first-party data accessible and actionable. As a Senior AI Engineer, you will design and build scalable backend systems and user experiences that power AI products and solutions, collaborating closely with product managers, machine learning engineers, and data scientists to drive impactful results.

Responsibilities:

• Design and build backend systems that support scaling our AI solutions for 167K+ customers
• Develop robust, reliable and scalable data collection and processing pipelines for machine learning models to train and consume
• Develop robust, reliable and scalable services to serve AI models in production environments
• Contribute to evolving our agentic architecture — making our AI agents more self-sufficient and performant
• Contribute to a culture of ownership, experimentation, and customer-centric product thinking

Requirements:

• 5+ Have hands-on experience building and deploying generative AI and agentic AI applications into production, with expertise in prompt engineering, few-shot learning, fine tuning and evaluation
• Experienced backend engineer with a strong track record of building scalable, distributed systems, especially in the service of AI agent capabilities
• Proficient in Python and modern backend frameworks (FastAPI, Django preferred)
• Experience creating human and automated evals to ensure high AI model quality
• Proficient in big data tools such as Apache Spark and Hadoop
• Deep experience with asynchronous processing and distributed task queues (Celery, Kafka, SQS, RabbitMQ, Redis)
• Strong understanding of database technologies and ORMs (SQLAlchemy, Alembic)
• Comfortable with cloud-native architectures (AWS) and container orchestration (Kubernetes); can manage infrastructure and CI/CD pipelines
• Skilled at designing and building robust APIs
• Able to operate with a high degree of autonomy, handle ambiguity, and thrive in a fast-moving, startup-like environment
• Driven by curiosity. You're someone who learns and stays up to date with this rapidly evolving field
• Comfortable collaborating directly with product managers and customers to shape solutions
• Trained ML models in the past and deployed them in production systems to generate impact to businesses
• Experience in reinforcement learning
• Contributions to open-source AI frameworks