✦ Luna Orbit — AI & Machine Learning

Senior Staff Platform Infrastructure Engineer

at Braze

📍 Chicago Onsite Posted April 01, 2026
Apply Now → ← Browse All Jobs

Position Details

Type Full-Time
Experience mid
Exp. Years 2+ years
Education Not specified
Category AI & Machine Learning

About this role

Software Engineer II, Machine Learning at Braze is focused on designing, implementing, and deploying data-intensive ML pipelines and AI-native infrastructure. You will work with Python-based data/ML ecosystems and production-grade cloud tooling.

Key Responsibilities

  • Use robust software engineering practices to design, implement, and improve modular components in a cutting-edge ML product
  • Work with customers to translate use cases into platform components
  • Apply Python and ML tooling to production pipelines
  • Collaborate on roadmap with Product/Design/Data
  • Ensure quality and reliability of production ML systems

Technical Overview

Data/ML stack includes Python, Pyspark, Polars, Ibis, SQL/BigQuery, FastAPI; orchestration and devops via Kubernetes, Airflow, Terraform; cloud platform experience (GCP) required.

Ideal Candidate

The ideal candidate is a mid-level machine learning software engineer with 2+ years of Python in production, strong cloud platform experience (GCP/AWS/Azure), and a track record of putting ML models into production. you should be comfortable building production ML pipelines and collaborating with data scientists.

Must-Have Skills

2+ years of experience working with Python in a product settingExperience putting ML models into productionExperience with at least one major cloud platform (GCPAWSAzure)

Nice-to-Have Skills

Experience with production ML pipelinesExperience with PySpark and BigQuery

Tools & Platforms

PythonPysparkPolarsIbisBigQuerySQLFastAPIKubernetesAirflowTerraformGoogle Cloud Platform

Required Skills

2+ years of experience working with Python in a product settingexperience putting ML models into productionexperience with a major cloud platform (GCPAWSAzure)

Hard Skills

PythonPysparkPolarsIbisSQLBigQueryFastAPIKubernetesAirflowTerraformGoogle Cloud PlatformGCP

Soft Skills

communicationteamworkproblem-solvingownership

Industry & Role

Industry SaaS
Job Function Develop and scale Braze's self-learning AI platform and ML pipelines in production.
Role Subtype Platform Engineer
Tech Domains Python, SQL / PostgreSQL, Google Cloud Platform, Kubernetes, Airflow, Terraform, BigQuery, Pyspark, Polars, Ibis, FastAPI

Keywords for Your Resume

Software Engineer IIMachine LearningBrazePythonPysparkPolarsIbisBigQuerySQLFastAPIKubernetesAirflowTerraformGoogle Cloud PlatformGCPML productiondata sciencesupervised learningproduction ML pipelines

Deal Breakers

No production Python experience, No cloud platform experience, Unwilling to work onsite in Toronto

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile