# Senior Software Engineer, ML/AI Platform

**Company:** [Attentive](https://hotfix.jobs/companies/attentive)
**Location:** Remote
**Role:** ML Engineering
**Salary:** $220k – $260k/yr
**Experience:** 5+ years
**Skills:** Spark, Spark Streaming, Ray Data, Kubernetes, Aws Eks, Python, PyTorch, TensorFlow, Metaflow, Kafka
**Posted:** 2026-05-19

> Senior Software Engineer on the ML Platform team building and operating data, tooling, serving, and inference layers for a PB-scale feature store and ML lifecycle at a high-growth AI marketing company.

## Job Description

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

- 5+ years working in Data Engineering / MLOps, with experience building and maturing pipelines of a PB-scale feature store.
- Deep experience with Apache Spark, Spark Streaming, and Ray Data for building data pipelines for ML use cases.
- Understanding of the correlation between data cardinality, query plans, configuration settings, hardware, and their impact on data pipeline performance.
- Experience creating infrastructure for training ML models and fine-tuning LLMs.
- Understanding of key differences between online and offline ML inferences and critical elements for success with each.

## What We Use

- Infrastructure runs primarily in Kubernetes hosted in AWS’s EKS.
- Infrastructure tooling includes Istio, Datadog, Terraform, CloudFlare, and Helm.
- Backend: Java / Spring Boot microservices, built with Gradle, coupled with DynamoDB, Kinesis, AirFlow, Postgres, Planetscale, and Redis, hosted via AWS.
- Frontend: React and TypeScript, using GraphQL, Storybook, Radix UI, Vite, esbuild, and Playwright.
- Automation: Python, Metaflow, HuggingFace, PyTorch, TensorFlow, and Pandas.

## Similar roles

- [Lead Software Engineer, Data Platform](https://hotfix.jobs/jobs/971c7657-3720-45b5-895f-92b209c5be4c) - Viam - New York, NY - $220k – $250k/yr
- [Senior Software Engineer, Machine Learning (Ads)](https://hotfix.jobs/jobs/1db54550-6fc7-466f-ac24-381a760fa14c) - Discord - San Francisco, CA - $220k – $247k/yr
- [Senior Machine Learning Engineer](https://hotfix.jobs/jobs/6fc6255d-fb31-4ef8-9926-6151402c0eaf) - Bluesky Social - Remote - $221k – $405k/yr
- [Senior Machine Learning Engineer, Ads Content Understanding](https://hotfix.jobs/jobs/ada32c3a-a5d5-401a-806b-70910c5055b4) - Reddit - Remote - $217k – $303k/yr
- [Senior Machine Learning Systems Engineer](https://hotfix.jobs/jobs/75802f62-e943-47ee-b612-02a773255c04) - Reddit - Remote - $217k – $303k/yr

**Apply:** https://hotfix.jobs/jobs/1b7a2837-b605-4dd4-b03e-aff6f8c85440
**Canonical:** https://hotfix.jobs/jobs/1b7a2837-b605-4dd4-b03e-aff6f8c85440