Senior Machine Learning Platform Engineer, Feature Store Developer

Attentive; United States; Remote; USD 144,000 - 210,000; Full time

Job summary

An innovative company is seeking a Senior Software Engineer to join its Machine Learning Operations team. This role offers the chance to influence the architectural roadmap and enhance the ML platform that powers AI-driven solutions. You'll work with cutting-edge technologies and collaborate with cross-functional teams to deliver impactful ML initiatives. This dynamic environment fosters growth and encourages you to take ownership of your projects. If you're passionate about data engineering and MLOps, this is an exciting opportunity to make a significant impact in a rapidly evolving field.

Benefits

Health & Wellness Benefits

Equity Options

Flexible Work Environment

Professional Development Opportunities

Qualifications

• 5+ years in Data Engineering/MLOps with PB-scale feature store experience.
• Deep experience with Apache Spark and building data pipelines for ML.

Responsibilities

• Unlock offline & real-time access to trillions of data points.
• Manage and optimize our feature store for ML and Data Science teams.

Skills

Data Engineering

MLOps

Apache Spark

Feature Engineering

Data Pipeline Optimization

Machine Learning

Tools

Kubernetes

AWS

Terraform

Java

Spring Boot

React

TypeScript

Python

Metaflow

PyTorch

TensorFlow

Job description

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 looking for a self-motivated, highly driven Senior Software Engineer to join our Machine Learning Operations (MLOps) team. As a team, we enable Attentive’s Machine Learning (ML) practice to directly impact Attentive’s AI product suite through the tools to train, inference, and deploy ML models with higher velocity and performance, while maintaining reliability. We build and maintain a foundational ML platform spanning the full ML lifecycle for consumption by ML engineers and data scientists. This is an exciting opportunity to join a rapidly growing MLOps team at the ground floor with the ability to drive and influence the architectural roadmap enabling the entire ML organization at Attentive.

This team and role is responsible for building and operating the ML data, tooling, serving, and inference layers of the ML platform. We are excited to bring on more engineers to continue expanding this stack.

What You'll Accomplish

• Unlock offline & real-time access to trillions of data points for our ML and Data Science teams
• Manage, expand, and optimize our feature store that enables feature engineering, multi-TB scale training jobs, and offline / real-time inferencing
• Support PB scale data operations on the feature store using Apache Spark, Spark Structured Streaming, Kafka, and Ray
• Partner with other teams and business stakeholders to deliver ML and AI initiatives

Your Expertise

• You have been working in the areas of Data Engineering / MLOps for 5+ years, and have built and matured the pipelines of a PB-scale feature store
• You have deep Apache Spark, Spark Streaming, and Ray Data experience and built data pipelines for ML use cases using these tools
• You understand the correlation between data cardinality, query plans, configuration settings, and hardware and the impact of each on data pipeline performance
• You have led the rollout and operationalization of feature stores such as Tecton or Feast
• You understand the key differences between online and offline ML inferences and can voice the critical elements to be successful with each to meet business needs

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 $144,840 - $210,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-EZ1

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.