Staff Software Engineer, Machine Learning

Attentive

New York (NY)

On-site

USD 320,000 - 360,000

Full time

2 days ago
Be an early applicant
Application generator

A complete application in a minute — tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Health & wellness
Equity

Job summary

Attentive is hiring a Senior Machine Learning Engineer to build, scale, and operate production-grade ML systems enabling real-time personalization across the Attentive platform.

You'll partner with Product and Engineering, own critical ML components, and raise the technical bar in a fast-paced, late-stage startup environment. Strong experience with TensorFlow/PyTorch, Python, and large-scale data analytics is required.

Qualifications

  • 6+ years building ML systems
  • Experience with TensorFlow/PyTorch, xgboost, pandas, matplotlib, SQL, Spark or similar tools
  • Proficiency in Python
  • Extensive experience using ML and data analysis to build scalable data-driven products
  • Led cross-functional ML projects across teams
  • Strong communication and collaboration skills

Responsibilities

  • Build and scale production-grade ML systems
  • Operate ML components with ownership and accountability
  • Collaborate with Product and Engineering to raise the technical bar
  • Design and implement data-driven features for real-time personalization
  • Drive cross-functional ML projects from concept to deployment
  • Mentor and influence engineering teams across ML initiatives

Skills

Python
ML systems design
Cross-functional leadership
Big data analytics
Single-system ownership
Collaboration & communication

Tools

TensorFlow/PyTorch
xgboost
pandas
matplotlib
SQL
Spark

Job description

Attentive® is the AI marketing platform for 1:1 personalization redefining the way brands and people connect. We’re the only marketing platform that combines powerful technology with human expertise to build authentic customer relationships. By unifying SMS, RCS, email, and push notifications, our AI-powered personalization engine delivers bespoke experiences that drive performance, revenue, and loyalty through real-time behavioral insights.

Recognized as the #1 provider in SMS Marketing by G2, Attentive partners with more than 8,000 customers across 70+ industries. Leading global brands like Crate and Barrel, Urban Outfitters, and Carter’s work with us to enable billions of interactions that power tens of billions in revenue for our customers.

With a distributed global workforce and employee hubs in New York City, San Francisco, London, and Sydney, Attentive’s team has been consistently recognized for its performance and culture. We’re proud to be included in Deloitte’s Fast 500 (four years running!), LinkedIn’s Top Startups, Forbes’ Cloud 100 (five years running!), Inc.’s Best Workplaces, and the Human Rights Campaign Foundation's Corporate Equality Index!

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

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

Attentive Company Values

Default to Action - Move swiftly and with purpose

Be One Unstoppable Team - Rally as each other’s champions

Champion the Customer - Our success is defined by our customers' success

Act Like an Owner - Take responsibility for Attentive’s success

Learn more about AWAKE, Attentive’s collective of employee resource groups.

If you do not meet all the requirements listed here, we still encourage you to apply! No job description is perfect, and we may also have another opportunity that closely matches your skills and experience.

At Attentive, we know that our Company's strength lies in the diversity of our employees. Attentive is an Equal Opportunity Employer and we welcome applicants from all backgrounds. Our policy is to provide equal employment opportunities for all employees, applicants and covered individuals regardless of protected characteristics. We prioritize and maintain a fair, inclusive and equitable workplace free from discrimination, harassment, and retaliation. Attentive is also committed to providing reasonable accommodations for candidates with disabilities. If you need any assistance or reasonable accommodations, please let your recruiter know.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff Software Engineer, Machine Learning
Staff Software Engineer, Machine Learning

Attentive • San Francisco (CA)

On-site
USD 320,000 - 360,000
Health & wellness
Equity
Senior Machine Learning Engineer
Senior Machine Learning Engineer

Attentive • New York (NY)

On-site
USD 244,000 - 320,000
Health benefits
Equity
Wellness programs
Software Engineer II, Segment Personalization
Software Engineer II, Segment Personalization

Attentive • United States

On-site
USD 150,000 - 190,000
Health benefits
Equity
Professional growth
Staff Software Engineer, Shopper Data Platform
Staff Software Engineer, Shopper Data Platform

Attentive • Wilmington (DE)

On-site
USD 180,000 - 230,000
Health & wellness
Equity
Senior Software Engineer, Onsite Customer Growth
Senior Software Engineer, Onsite Customer Growth

Attentive • New York (NY)

Hybrid
USD 170,000 - 230,000
Equity
Health benefits
Hybrid work model
Staff Software Engineer, Shopper Data Platform
Staff Software Engineer, Shopper Data Platform

Attentive • United States

On-site
USD 180,000 - 230,000
Health benefits
Equity
Senior Software Engineer
Senior Software Engineer

Attentive • New York (NY)

Hybrid
USD 132,000 - 230,000
Health & wellness
Equity
Staff Software Engineer, Onsite Customer Growth
Staff Software Engineer, Onsite Customer Growth

Attentive • New York (NY)

Hybrid
USD 205,000 - 275,000
Health & wellness benefits
Equity
Software Engineer II, Integrations
Software Engineer II, Integrations

Attentive • Wilmington (DE)

On-site
USD 135,000 - 170,000
Health insurance
Equity
Staff Software Engineer, Shopper Data Platform
Staff Software Engineer, Shopper Data Platform

Attentive • San Francisco (CA)

On-site
USD 180,000 - 230,000
Equity
Health benefits
Wellness programs