Staff Software Engineer, ML/AI Platform

Attentive
United States
Workplace: OnsiteFull timeUSD 200,000 - 300,000 annuallyFunction: Data Science & Machine LearningSkills: ["Technical leadership","Mentorship","Architectural decision-making","Influence","High-leverage problem solving"]

Architect and build the foundational AI/ML platform capabilities that power model training, deployment, and real-time serving at massive scale. Set technical direction across ML platform strategy, low-latency inference, and robust model lifecycle management (versioning, rollouts, rollback, champion/challenger). Define and drive an agentic stack, create universal serving/data access interfaces, and mentor platform/ML engineers while partnering across multiple AI teams.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Attentive
Attentive
3 months ago

Staff Software Engineer, ML/AI Platform

âś“ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 3 hours agoStatus: Live

Job Summary

Architect and build the foundational AI/ML platform capabilities that power model training, deployment, and real-time serving at massive scale. Set technical direction across ML platform strategy, low-latency inference, and robust model lifecycle management (versioning, rollouts, rollback, champion/challenger). Define and drive an agentic stack, create universal serving/data access interfaces, and mentor platform/ML engineers while partnering across multiple AI teams.
Location: United States
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Architect ML platform strategy spanning data pipelines, training infrastructure, and serving layers using tooling like Ray, MLFlow, Metaflow, Argo, and Spark.
  • •Build and operate production-grade, low-latency ML serving layers with model lifecycle systems (champion/challenger testing, automated rollouts, versioning, rollback).
  • •Define and drive an agentic stack across Attentive’s AI organization.
  • •Provide ML infrastructure perspective in high-level discussions about AI strategy across multiple quarters and teams.
  • •Mentor platform and ML engineers and build universal interfaces and patterns (data access layers and prediction serving APIs) to streamline ML work across the organization.

Pay and Benefits

Salary: USD 200,000 - 300,000 annually
Perks:Equity

Key Requirements

  • •7+ years focused on ML Platform/MLOps with strong knowledge of best practices and tooling.
  • •Proven experience owning and building core ML platform components using tools such as Spark, Ray, MLFlow, Kubeflow, or Metaflow.
  • •Experience designing and operating online and offline inference systems, including key tradeoffs between them.
  • •Strong Python expertise for batch processing and online service frameworks.
  • •Experience building and operating a high-throughput agentic stack (MCP/data infrastructure, context store, orchestration, and prompt layer).
Experience:AI/MLMLOpsPlatform engineeringAgentic systems
Skills:Technical leadershipMentorshipArchitectural decision-makingInfluenceHigh-leverage problem solving
Languages:English
Tech Stack:RayMLFlowMetaflowArgoSparkKubeflowPythonJavaKotlinSpring BootGradleDynamoDBAuroraAirFlowPostgresRedisAWSKubernetesEKSIstio

Company Brief

Attentive
Provides a mobile messaging and personalization platform that helps brands engage customers via SMS, email, and onsite messaging to drive sales, retention, and personalized marketing at scale.
Industry: Enterprise Software
Company Size: Enterprise (1,001+ employees)
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: New York, United States
Founded: 2016
WebsiteLinkedIn