Senior Software Engineer, ML/AI Platform

About the Role

We’re looking for a self-motivated, highly driven Senior Software Engineer to join our Machine Learning Platform (MLPlatform) 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, serve, and deploy ML models with higher velocity and performance, while maintaining reliability. We build and maintain a foundational ML platform that spans the full ML lifecycle for use by ML engineers and data scientists. This is an exciting opportunity to join a rapidly growing ML Platform 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 are 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 know/have created infrastructure for Training ML models/fine-tuning LLMs.
• 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 $180,000 - $250,000 annually + equity + benefits. Our salary ranges are determined by role, level and location.

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

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.