Braze is looking for a Lead Machine Learning Engineer to lead a team building next-generation personalization and optimization systems. You will balance hands-on technical work with team leadership, driving both technical excellence and measurable business impact.
What You'll Do
- Lead a team of 3-5 ML engineers building personalization systems
- Architect multi-stage ranking systems (retrieval → ranking → re-ranking)
- Define technical roadmap aligned with business objectives
- Drive improvements in model quality, system reliability, and development velocity
- Establish best practices for model development and deployment
- Partner with product, data science, and engineering teams
- Contribute hands-on to critical model development (50% coding)
What We're Looking For
- 7+ years in ML engineering
- 2+ years leading technical teams
- Deep expertise in large-scale personalization systems
- Experience with modern ML architectures (two-tower, transformer-based, graph neural networks)
- Strong system design skills for ML infrastructure
- Proven ability to mentor engineers and drive technical decisions
- Experience managing stakeholder relationships and project delivery
- BS/MS in Computer Science or equivalent experience
Nice to Have
- Experience in adtech, e-commerce, or content platforms
- Published papers or significant ML contributions
- Experience with multi-objective optimization
- Knowledge of reinforcement learning for personalization
Team & Environment
You will lead a team of 3-5 ML engineers in a high-growth, high-performance culture where speed, ownership, and measurable impact drive success.
Benefits & Compensation
- Compensation: $130,000 - $250,000 per year
- Profit-sharing bonus
- Competitive benefits
Work Mode
This is a remote position. Candidates should be based in or near Riverside, CA.
We are proud to be an Equal Employment Opportunity company. We value diversity, equity, and inclusion. We do not discriminate based on race, religion, color, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics.

