Staff Software Engineer, Machine Learning

Attentive · United States · Engineering

Posted 2026-05-12

Apply for this role →

About the Role

Our Machine Learning Engineering team powers personalized experiences for hundreds of millions of customers across thousands of brands. As a Senior Machine Learning Engineer, you will play a critical role in building, scaling, and operating production-grade ML systems that drive real-time personalization across the Attentive platform.

You will operate with a high degree of ownership, partner closely with Product and Engineering, and help raise the technical bar across our ML systems in a fast-paced, high-impact environment.

What You’ll Accomplish

You have a proven track record of building systems that maintain a high bar of quality

You deeply loathe regressions and take proactive steps to protect against them through a variety of testing techniques

You are a collaborator, technical leader, and a great communicator

You are constantly improving the quality of the project you are working on, both via direct contributions as well as long-term advocacy for larger-scale changes

You are enthusiastic about the high impact, fast-paced work environment of an late-stage startup

10+ years experience is ideal

Your Expertise

You have worked professionally building systems for 6+ years with experience on a single system long enough to see the consequences of your decisions

Experience with TensorFlow/Pytorch, xgboost, pandas, matplotlib, SQL, Spark or similar tools

You have proficiency or experience with Python

You have extensive experience using machine learning and data analysis, or similar, to build scalable systems and data-driven products, working with cross-functional teams

You have a proven track record of building scalable, efficient, automated processes for large-scale data analyses, model development, model validation, and model implementation from modern research

You have led cross-functional machine learning projects across teams

What We Use

Our infrastructure runs primarily in Kubernetes hosted in AWS’s EKS

Infrastructure tooling includes Istio, Datadog, Terraform, CloudFlare, and Helm

Our backend is Java / Spring Boot microservices, built with Gradle, coupled with things like DynamoDB, Kinesis, AirFlow, Postgres, Planetscale, and Redis, hosted via AWS

Our frontend is built with React and TypeScript, and uses best practices like GraphQL, Storybook, Radix UI, Vite, esbuild, and Playwright

Our automation is driven by custom and open source machine learning models, lots of data and built with Python, Metaflow, HuggingFace 🤗, PyTorch, TensorFlow, and Pandas

You'll get competitive perks and benefits, from health & wellness to equity, to help you bring your best self to work.

For US based applicants:

The US base salary range for this full-time position is $320,000 - $360,000 annually + equity + benefits

Our salary ranges are determined by role, level and location

#LI-EF1

By applying for this position, your data will be processed as per Attentive's Privacy Policy.

Apply for this role →

← Back to all jobs