SlipstreamJobs tracks this role from the company's public career site. Apply directly on the employer's site.
Salary: USD 184,000 - 314,000 / annual
Braze is seeking a Staff Machine Learning Engineer to join the Predictive and Generative AI (PGAI) team. The team's mission is to deliver engaging and personalized customer experiences through ML and AI-enhanced marketing solutions, running those systems at global scale with distributed training pipelines and high-throughput APIs serving predictions across multiple regions.
In this role, you will own the platform infrastructure underneath ML operations and make deploying, operating, and scaling ML at Braze fast, safe, and efficient. You will identify and drive transformative initiatives that change how the team runs ML in production—whether replatforming queueing and orchestration, overhauling deployment and cloud identity, or retiring infrastructure generations. This is a hands-on delivery role where you carry the most complex infrastructure initiatives yourself from design through production, including multi-region model serving fleets, pipelines that keep hundreds of customer-specific models healthy, and CI/deployment tooling that moves it all safely.
You will own the platform's technical vision and production quality bar, setting direction for how models are trained, deployed, served, and observed. You'll lead incident response for ML systems and drive reliability and cost work that keeps the platform efficient at scale. You will drive initiatives that span teams, building on shared infrastructure and data systems owned with partner teams, and carry technical relationships with those teams. You'll raise the team's engineering quality through design review, code review, and production readiness for ML systems, mentoring other senior engineers and data scientists. You'll connect technical decisions to customer and business outcomes and represent the team's technical perspective to product and engineering leadership.
Braze is the leading customer engagement platform empowering brands to deliver great customer experiences. The company has 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 named a 2025 Best Companies To Work For by U.S. News & World Report and a 2025 Fortune Best Workplace in Technology by Great Place To Work, among other accolades.
REQUIREMENTS:
- 8+ years building and operating distributed systems in production, with depth in deployment and operations. You must have designed services for scale and reliability, owned CI/CD and infrastructure as code, and run what you built under production load.
- Hands-on experience with ML workloads in production. Training pipelines, model serving, feature systems, or ML platform tooling all count; deep modeling experience is a plus rather than a requirement.
- A technical leader who has owned direction for a team, led multi-quarter initiatives across team boundaries, and grown senior engineers, all while keeping high personal output.
- Deep working knowledge of Kubernetes and cloud infrastructure, including identity and access management, networking, and cost profile of what you run.
- An effective communicator, both verbal and written, whose designs and recommendations build consensus and drive forward decision making.
BONUS QUALIFICATIONS:
- Experience with queueing and orchestration systems such as Celery, RabbitMQ, Kafka, or Ray.
- ML platform tooling such as MLflow or another model registry, feature stores, or ML observability.
- Experience in Braze's stack (Python, Ruby on Rails, MongoDB, Redis, Kubernetes).
- Operating under compliance regimes such as SOX or HIPAA.
- Customer engagement, personalization, or marketing technology domain experience.
