Skip to content
AttentiveAttentiveUnited States

Senior Software Engineer, ML/AI Platform

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.

220k – 260k/yr
Remote5+ YOEML Engineering

About the role

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.

Skills

SparkSpark StreamingRay DataKubernetesAws EksPythonPyTorchTensorFlowMetaflowKafka

Similar roles

ML Engineering jobs
Viam

Lead Software Engineer, Data Platform

ViamNew York, NY

Viam is seeking a Lead Software Engineer to lead the Data/ML team, owning technical direction, architecture, and delivery. This hands-on role involves managing a team of 5+ engineers, writing code, and driving the reliability and performance of ML training and inference infrastructure.

220k – 250k/yr
HybridML Engineering
Discord

Senior Software Engineer, Machine Learning (Ads)

DiscordSan Francisco, CA

Develops and deploys ML models for ads targeting, ranking, and measurement on Discord's Ads platform. Requires 5+ years ML experience, 3+ years in Ads ML, Python proficiency, and PyTorch/TensorFlow expertise.

220k – 247k/yr
On-site5+ YOEML Engineering
Bluesky Social

Senior Machine Learning Engineer

Bluesky SocialSeattle, WA

Designs, develops, and maintains ML systems for content recommendations, search, spam detection, and labeling using Bluesky's social graph. Requires 3+ years in ML/data science focused on recommendations/search, Python/PyTorch proficiency, and rapid experimentation skills.

221k – 405k/yr
Remote3+ YOEML Engineering
Reddit

Senior Machine Learning Engineer, Ads Content Understanding

RedditUnited States

Senior Machine Learning Engineer building and shipping production ML systems for Ads Content Understanding at Reddit. Focus on LLMs, content taxonomies, opinion mining, product understanding, and operationalizing signals for monetization impact. Requires 5+ years delivering scalable ML in production.

217k – 303k/yr
Remote5+ YOEML Engineering
Reddit

Senior Machine Learning Systems Engineer

RedditUnited States

Build large-scale ML experimentation and training orchestration platforms, including agentic AI execution systems, to accelerate Ads ML development at Reddit. Requires 5+ years infrastructure experience and 2+ years building production ML platforms.

217k – 303k/yr
Remote5+ YOEML Engineering