AI Platform Engineer II

At Braze, we have found our people. We’re a genuinely approachable, exceptionally kind, and intensely passionate crew. We seek to ignite that passion by setting high standards, championing teamwork, and creating work-life harmony as we collectively navigate rapid growth on a global scale while striving for greater equity and opportunity – inside and outside our organization. To flourish here, you must be prepared to set a high bar for yourself and those around you. There is always a way to contribute: Acting with autonomy, having accountability and being open to new perspectives are essential to our continued success. Our deep curiosity to learn and our eagerness to share diverse passions with others gives us balance and injects a one-of-a-kind vibrancy into our culture. If you are driven to solve exhilarating challenges and have a bias toward action in the face of change, you will be empowered to make a real impact here, with a sharp and passionate team at your back. If Braze sounds like a place where you can thrive, we can’t wait to meet you. WHAT YOU'LL DO Join the AI Platform team at Braze to build and scale BrazeAI Decisioning Studio - a reinforcement learning platform at the forefront of AI Decisioning. The platform runs continuous experimentation and personalizes customer engagement at the individual level, helping brands move from rule-based campaigns to autonomous, self-optimizing interactions. You'll work at the intersection of cloud-native infrastructure, data-intensive systems, and machine learning in production. Main responsibilities: Build and maintain critical services and subsystems on our AI platform, balancing performance with cost-effective operations Implement cloud-native solutions that ensure reliability, scalability, and fault tolerance Troubleshoot production incidents end-to-end, going deep to identify root causes and implement durable fixes Contribute to observability practices using Sentry and Datadog to proactively detect issues and minimize downtime Collaborate with data scientists, ML engineers, and product teams to translate real-world use cases into platform capabilities Improve developer experience by streamlining workflows, enhancing tooling, and supporting MLOps best practices Tech stack: Core Data & ML: Python, Ibis, FastAPI, Dataproc (Spark), SQL, BigQuery, MLflow, Streamlit Platform & Infrastructure: Google Cloud Platform, AWS, Kubernetes, Helm, Terraform Workflows & Orchestration: Airflow, RabbitMQ, Celery CI/CD: GitHub Actions, Jenkins Observability: Sentry, Datadog Why this role: Production ML at scale: no toy datasets or notebook demos; you’re building infrastructure that powers real AI workloads Engineering rigor: unit and integration tests, modular design, CI/CD, pair programming, and code reviews are how we work, not aspirations Learn continuously: deep exposure to ML system architecture, end-to-end ML workflows, and reinforcement learning systems WHO YOU ARE 2-4 years of experience in platform engineering, infrastructure, or a related backend role Solid understanding of platform architecture, particularly in ML or data-intensive environments Hands-on experience with Kubernetes and cloud infrastructure (GCP preferred) Ability to troubleshoot complex distributed systems under pressure Writes clean, modular code with a focus on testable APIs and maintainable design Experience working with AI coding assistants. Understands effective prompting strategies and can articulate when these tools add value versus when they're not appropriate Clear communicator who can work across technical and non-technical stakeholders Proactive problem solver who identifies issues and works around obstacles without waiting for direction For candidates based in Ontario, the pay range for this position at the start of employment is expected to be between CA$125,280 and CA$235,736/year with an expected On Target Earnings (OTE) between CA$139,200 and CA$258,000/year (including bonus or commission). Your exact offer may vary depending on multiple individualized factors, including market location, job-related knowledge, skills, and experience. In addition to cash compensation, Braze offers full- and part- time employees a comprehensive Total Rewards package that includes equity grants of restricted stock (RSUs) so that all Braze employees own a piece of our company. #LI-Hybrid WHAT WE OFFER Braze benefits vary by location, and we encourage you to review our specific benefits offerings for each country here. More details on benefits plans will be provided if you receive an offer of employment. From offering comprehensive benefits to fostering hybrid ways of working, we’ve got you covered so you can prioritize work-life harmony. Braze offers benefits such as: Competitive compensation that may include equity Retirement and Employee Stock Purchase Plans Flexible paid time off Comprehensive benefit plans covering medical, dental, vision, life, and disability Family services that include fertility benefits and equal paid parental leave Professional development supported by formal career pathing, learning platforms, and a yearly learning stipend A curated in-office employee experience, designed to foster community, team connections, and innovation Opportunities to give back to your community, including an annual company-wide Volunteer Week and donation matching Employee Resource Groups that provide supportive communities within Braze Collaborative, transparent, and fun culture recognized as a Great Place to Work® ABOUT BRAZE Braze is the leading customer engagement platform that empowers brands to Be Absolutely Engaging.™ Braze helps brands deliver great customer experiences that drive value both for consumers and for their businesses. Built on a foundation of composable intelligence, BrazeAI™ allows marketers to combine and activate AI agents, models, and features at every touchpoint throughout the Braze Customer Engagement Platform for smarter, faster, and more meaningful customer engagement. From cross-channel messaging and journey orchestration to Al-powered decisioning and optimization, Braze enables companies to turn action into interaction through autonomous, 1:1 personalized experiences. The company has repeatedly been recognized as a Leader in marketing technology by industry analysts, and was voted a G2 “Best of Marketing and Digital Advertising Software Product” in 2025. Braze was also named a 2025 Best Companies To Work For by U.S. News & World Report, a 2025 America’s Greatest Companies by Newsweek, and a 2025 Fortune Best Workplace in Technology™ by Great Place To Work®, among other accolades. Braze is also proudly certified as a Great Place to Work® in the U.S., the UK, Australia, and Singapore. The company is headquartered in New York with offices in Austin, Berlin, Bucharest, Chicago, Dubai, Jakarta, London, Paris, San Francisco, São Paulo, Singapore, Seoul, Sydney and Tokyo. BRAZE IS AN EQUAL OPPORTUNITY EMPLOYER At Braze, we strive to create equitable growth and opportunities inside and outside the organization. Building meaningful connections is at the heart of everything we do, and that includes our recruiting practices. We're committed to offering all candidates a fair, accessible, and inclusive experience – regardless of age, color, disability, gender identity, marital status, maternity, national origin, pregnancy, race, religion, sex, sexual orientation, or status as a protected veteran. When applying and interviewing with Braze, we want you to feel comfortable showcasing what makes you you. We know that sometimes different circumstances can lead talented people to hesitate to apply for a role unless they meet 100% of the criteria. If this sounds familiar, we encourage you to apply, as we’d love to meet you. Please see our Candidate Privacy Policy for more information on how Braze processes your personal information during the recruitment process and, if applicable based on your location, how you can exercise any privacy rights.