Senior ML Infrastructure Engineer

ironSource; Mountain View (WY); On-site; USD 140,000 - 232,000; Full time

Benefits offered by this job

Comprehensive health insurance

Employee stock ownership

Generous vacation and personal days

Job summary

Unity's ML Engineering Infrastructure team builds the platform that powers machine learning across the company, from feature generation and training data pipelines to distributed training, experimentation, and model serving. We're looking for a Machine Learning Engineer to help build the platform behind our product intelligence, production ML models, and the teams that develop them.

You will operate at scale across batch and streaming data, and collaborate with engineers across the org to

Qualifications

• Experience building ML infrastructure, data platforms, or distributed systems in production.
• Strong Python skills and experience working with data-intensive workloads.
• Hands-on experience with ML frameworks (e.g., PyTorch) and distributed compute (e.g., Ray, Spark)
• Experience with data pipelines, model training workflows, or large datasets beyond academic settings
• Comfortable working across several layers of the ML stack, with a track record of learning new tools and systems quickly
• Ability to spot patterns in ambiguous or inconsistent problems and translate them into practical technical needs
• Clear communication and a collaborative approach with peers, managers, and partner teams
• Bachelor's degree in Computer Science, Machine Learning, Systems, or a related field, or equivalent practical experience

Responsibilities

• Build and maintain data and feature pipelines that generate training datasets for ML models and experimentation
• Develop and improve infrastructure for distributed training workflows
• Build and operate multi-stage ML pipelines using workflow orchestration tools
• Contribute to model serving and deployment infrastructure, helping models move reliably from training into production
• Improve reproducibility and reliability across the stack through monitoring, alerting, dataset validation, and testing
• Take on projects scoped with your manager and carry them through end-to-end, from design to rollout, using your own judgment along the way
• Investigate uncommon or inconsistent issues across pipelines and training systems, recognize patterns, and turn them into concrete improvements
• Keep both the hands-on details and the bigger picture in mind, delivering on immediate needs while moving the platform in the right direction
• Partner with ML engineers, researchers, and peers across teams, and share progress and tradeoffs with managers and directors
• Pick up new tools and new parts of the stack as team priorities shift

Skills

Python

PyTorch

Ray

Spark

Data pipelines

Distributed systems

Education

Bachelor's degree in Computer Science, Machine Learning, Systems, or related field

Tools

Kubernetes

Prometheus

Grafana

Airflow

Prefect

Triton

KServe

Ray Serve

Kafka

Job description

Unity's ML Engineering Infrastructure team builds the platform that powers machine learning across the company, from feature generation and training data pipelines to distributed training, experimentation, and model serving. We're looking for a Machine Learning Engineer to help build the platform behind our product intelligence, production ML models, and the teams that develop them.

You will operate at scale across batch and streaming data, and collaborate with engineers across the org to