Staff Software Engineer, ML Platform

Posted 6 Days Ago
San Francisco, CA
Hybrid
215K-290K
Senior level
Artificial Intelligence • Marketing Tech • Mobile • Software
Attentive is the AI marketing platform for leading brands.
The Role
As a Staff Software Engineer, you'll architect and build the ML platform infrastructure, guiding AI initiatives and mentoring engineers while enhancing the AI/ML stack for scalable performance.
Summary Generated by Built In
Attentive® is the AI-powered mobile marketing platform transforming the way brands personalize consumer engagement. Attentive enables marketers to craft tailored journeys for every subscriber, driving higher recurring revenue and maximizing campaign performance. Activating real-time data from multiple channels and advanced AI, the platform personalizes content, tone, and timing to deliver 1:1 messages that truly resonate.

With a top-rated customer success team recognized on G2, Attentive partners with marketers to provide strategic guidance and optimize SMS and email campaigns. Trusted by leading global brands like Neiman Marcus, Samsung, Wayfair, and Dyson, Attentive ensures enterprise-grade compliance and deliverability, supporting trillions of interactions across more than 70 industries. To learn more or request a demo, visit www.attentive.com or follow us on LinkedIn, X (formerly Twitter), or Instagram.

Attentive’s growth has been recognized by Deloitte’s Fast 500, Linkedin’s Top Startups and Forbes Cloud 100 all thanks to the hard work from our global employees!

About the Role
We’re seeking an accomplished Staff Software Engineer to join Attentive’s Machine Learning Platform team as a high-impact individual contributor focused on building the AI and ML infrastructure that powers our AI product suite. You’ll architect and build the foundational platform components that enable AI / ML engineers and data scientists to train, deploy, and serve models and agentic infrastructure with velocity, performance, and reliability at scale.

As a Staff-level IC, you’ll operate as a technical force multiplier, setting the technical direction for AI and ML infrastructure across Attentive’s AI organization. You’ll lead through influence and technical excellence, advocating for long-term architectural progress while balancing immediate platform needs. Your work will span strategic initiatives measured in quarters and years, focusing on high-leverage decisions that enable entire teams to ship AI and ML capabilities faster and more reliably.

Strategic Need
Attentive is revolutionizing the digital shopping experience across every channel through our AI product suite for half a billion subscribers. We’re looking for a high impact individual to take our platform from v1 to vNext and beyond — supporting the full spectrum of AI and ML workloads at massive scale. We support traditional models and deep learning today, and we are growing into reinforcement learning and agentic infrastructure quickly. This is a ground-floor opportunity to drive and influence the architectural roadmap for Attentive’s entire AI and ML ecosystem toward self-service workflows, real-time inference at scale, agentic capabilities, and robust model lifecycle management. 

What You’ll Accomplish

  • Setting Technical Direction - Architect ML platform strategy spanning data pipelines, training infrastructure, and serving layers using cutting-edge tooling like Ray, MLFlow, Metaflow, Argo, and Spark
  • Uplevel and Innovate Core AI & ML Stack - Build and operate production-grade, low-latency ML serving layers with  robust model lifecycle systems including champion/challenger testing, automated rollouts, versioning, and rollback capabilities
  • Uplevel and Innovate Core AI & ML Stack - Define and drive Attentive’s agentic stack
  • Technical Leadership - Provide ML infrastructure perspective in high-level discussions about Attentive’s AI strategy spanning multiple quarters and teams
  • Technical Mentorship - Mentor platform and ML engineers, actively championing team members
  • Being the “Glue” - Build universal interfaces, architectures, and patterns—like data access layers and prediction serving APIs—that bridge platform capabilities with product needs to streamline high-priority ML work across the organization

Your Expertise

  • You have the experience to know what works, what doesn’t, and why in AI and ML systems
  • 5+ years focused specifically on ML Platform/MLOps, with deep understanding of gold-standard practices and best-in-class tooling
  • Proven track record of owning and building core components of ML platforms using tools like Spark, Ray, MLFlow, Kubeflow, or Metaflow
  • You’ve built and operated a high-throughput agentic stack (MCP / data infrastructure, context store, orchestration, and prompt layer)
  • Strong expertise in Python for both batch processing and online service frameworks
  • Experience designing and operating online and offline inference systems, understanding the critical differences and tradeoffs between them

Sample Projects

  • Design and implement inference pipelines with champion/challenger shadow testing and automated model promotion
  • Lead and scale Attentive’s agentic stack from the ground up
  • Scale real-time feature streaming to handle low-latency, high-volume reinforcement learning workloads
  • Build a universal data access layer and prediction serving interface that powers ML capabilities across Attentive’s product suite

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 $215,000 - $270,000 annually + equity + benefits
- Equity is a substantial part of the total compensation package
- Our salary ranges are determined by role, level and location

#LI-MDK1

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. 

Top Skills

Argo
Kubeflow
Metaflow
Mlflow
Python
Ray
Spark
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: New York, NY
1,000 Employees
Year Founded: 2016

What We Do

Attentive® is the AI marketing platform for leading brands, designed to optimize message performance through 1:1 SMS and email interactions. Infusing intelligence at every stage of the consumer’s purchasing journey, Attentive empowers businesses to achieve hyper-personalized communication with their customers on a large scale. Leveraging AI-powered tools, a mobile-first approach, two-way conversations, and enterprise-grade technology, Attentive drives billions in online revenue for brands around the globe. Trusted by over 8,000 leading brands such as CB2, Urban Outfitters, GUESS, Dickey’s Barbeque Pit, and Wyndham Resort, Attentive is the go-to solution for delivering powerful commerce experiences for consumers with the brands they love.

To learn more about Attentive or to request a demo, visit www.attentive.com or follow us on LinkedIn, X (formerly Twitter), or Instagram.

Why Work With Us

At Attentive, you'll connect with inspiring, high-caliber people, and be encouraged to take risks, get creative, and think bigger. We're solving big problems for our customers, through our innovative AI solutions, giving employees the opportunity to thrive along the journey. The sky's the limit when it comes to what's possible.

Gallery

Gallery

Similar Jobs

Discord Logo Discord

Staff Software Engineer

Gaming • Machine Learning • Mobile • Software
In-Office
San Francisco, CA, USA
272K-306K
In-Office
Los Angeles, CA, USA

Pony.AI Logo Pony.AI

Staff Software Engineer

Artificial Intelligence • Software • Transportation
In-Office
Fremont, CA, USA
220K-285K Annually

Similar Companies Hiring

Credal.ai Thumbnail
Software • Security • Productivity • Machine Learning • Artificial Intelligence
Brooklyn, NY
Standard Template Labs Thumbnail
Software • Information Technology • Artificial Intelligence
New York, NY
10 Employees
PRIMA Thumbnail
Travel • Software • Marketing Tech • Hospitality • eCommerce
US
15 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account