Staff Software Engineer, Feature Platform & ML Infra

Unity; Mountain View (CA); On-site; USD 193,000 - 289,000; Full time

Benefits offered by this job

Equity / Stock ownership

Comprehensive health insurance

Generous vacation & personal days

Family leave support

Job summary

Unity is seeking a Staff Software Engineer for the Feature Platform to design and operate scalable data pipelines and feature datasets powering ads optimization and ML. You will own end-to-end offline data flows, ensure correctness, reproducibility, and SLA compliance, and collaborate with ML, backend, and product teams.

You'll build and maintain large-scale batch and streaming systems (Python/Java/SQL), contribute to the Feature Store, and improve observability, testing, and performance at

Qualifications

• Proven track record designing and operating large-scale distributed systems.
• Hands-on ETL/ELT pipelines using Python, Java, SQL, or similar technologies.
• Experience with distributed processing frameworks such as Spark or Flink in batch and streaming modes, including performance tuning.
• Understanding of how offline data systems integrate with online serving layers - feature stores, low-latency APIs, and real-time systems.

Responsibilities

• Design, build, and operate scalable, production-grade data pipeline systems and curated feature datasets powering ads optimization and ML.
• Own end-to-end offline data flows from raw event ingestion to feature-ready datasets, with strong emphasis on correctness, reproducibility, and SLA compliance.
• Develop and maintain large-scale batch and streaming systems (Python / Java / SQL) with a strong focus on performance, cost-efficiency, and reliability.
• Build and contribute to our Feature Store platform, including integration with the high-throughput online serving layer (Go-based services).
• Translate complex product and monetization logic into well-engineered, extensible systems serving analytics and machine learning use cases.
• Drive improvements in observability, testing frameworks, and quality standards across the platform.
• Lead architectural decisions and engineering best practices within the Feature Platform team.

Skills

Distributed systems design

ETL/ELT pipelines

Python/Java/SQL

Spark or Flink

Feature stores & online serving

Go programming

Kubernetes

Observability & reliability

Ownership mindset

Tools

Spark

Flink

Kubernetes

Go

Python

Java

SQL

Job description

Unity is seeking a Staff Software Engineer for the Feature Platform to design and operate scalable data pipelines and feature datasets powering ads optimization and ML. You will own end-to-end offline data flows, ensure correctness, reproducibility, and SLA compliance, and collaborate with ML, backend, and product teams.

You'll build and maintain large-scale batch and streaming systems (Python/Java/SQL), contribute to the Feature Store, and improve observability, testing, and performance at