Staff Software Engineer, ML Platform jobs in United States
cer-icon
Apply on Employer Site
company-logo

Attentive · 6 days ago

Staff Software Engineer, ML Platform

Attentive is the AI marketing platform for 1:1 personalization redefining the way brands and people connect. The Staff Software Engineer will focus on building the AI and ML infrastructure that powers the AI product suite, leading technical direction and mentoring engineers to enhance the platform's capabilities.

Artificial Intelligence (AI)Email MarketingMarketing AutomationMessagingMobilePersonalization
check
H1B Sponsor Likelynote

Responsibilities

Architect ML platform strategy spanning data pipelines, training infrastructure, and serving layers using cutting-edge tooling like Ray, MLFlow, Metaflow, Argo, and Spark
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
Define and drive Attentive’s agentic stack
Provide ML infrastructure perspective in high-level discussions about Attentive’s AI strategy spanning multiple quarters and teams
Mentor platform and ML engineers, actively championing team members
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

Qualification

ML Platform/MLOpsPythonSparkRayMLFlowMetaflowKubeflowReinforcement LearningTechnical LeadershipTechnical Mentorship

Required

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
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

Benefits

Equity
Benefits

Company

Attentive

company-logo
Attentive is a personalized mobile messaging platform that facilitates businesses with AI powered SMS and e-mail marketing solutions.

H1B Sponsorship

Attentive has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (33)
2024 (21)
2023 (10)
2022 (8)
2021 (8)
2020 (5)

Funding

Current Stage
Late Stage
Total Funding
$863M
Key Investors
CoatueSequoia CapitalBain Capital Ventures
2023-08-17Secondary Market
2021-03-24Series E· $470M
2020-09-23Series D· $230M

Leadership Team

leader-logo
Amit Jhawar
Chief Executive Officer
linkedin
leader-logo
Todd Berman
VP of Engineering
linkedin
Company data provided by crunchbase