Description
Braze is seeking a Staff Machine Learning Engineer to join our Predictive and Generative AI (PGAI) team. The team's mission is to deliver a truly engaging and personalized customer experience through the creation of ML and AI enhanced marketing solutions.
You will own the platform underneath, and you will make deploying, operating, and scaling ML at Braze fast, safe, and efficient. As the Staff Engineer on the team, you will:
- Identify and drive the transformative initiatives that change how the team runs ML in production
- Build and ship at high velocity, carrying the most complex infrastructure initiatives yourself from design through production
- Own the platform's technical vision and production quality bar, setting direction for how models are trained, deployed, served, and observed
- Drive initiatives that span teams, carrying the technical relationships with those teams
- Raise the team's engineering quality through design review, code review, and production readiness for ML systems, and mentor other senior engineers and data scientists
Requirements:
- 8+ years building and operating distributed systems in production, with depth in deployment and operations
- Hands-on experience with ML workloads in production
- A technical leader who has owned direction for a team, led multi-quarter initiatives across team boundaries, and grown senior engineers
- Deep working knowledge of Kubernetes and cloud infrastructure
- An effective communicator, both verbal and written
Bonus skills:
- 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 our 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
For candidates based in the United States, the pay range for this position at the start of employment is expected to be between $184,000 and $314,000/year, with an expected On Target Earnings (OTE) between $204,000 and $348,000/year.