[Remote] Staff Engineer, ML/AI Platform

Note: The job is a remote job and is reputed company to candidates in USA. reputed company is the AI marketing platform for 1:1 personalization redefining the way brands and people connect. They are seeking an accomplished Staff Software Engineer to join their Machine Learning Platform team, focusing on building the AI and ML infrastructure that powers their AI product suite.

Responsibilities

• 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
• reputed company and reputed company 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
• reputed company and reputed company Core AI & ML Stack - Define and drive reputed company’s agentic stack
• Technical Leadership - reputed company ML infrastructure perspective in high-level discussions about reputed company’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 reputed company layers and reputed company serving APIs—that reputed company platform capabilities with product needs to streamline high-reputed company ML work across the organization

Skills

• 7+ 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
• You have the experience to know what works, what doesn't, and why in AI and ML systems

Benefits

• Competitive perks and benefits, from health & wellness to equity, to help you bring your best self to work
• Equity
• Benefits

Company Overview

Company H1B Sponsorship