ML Infrastructure Engineer
New
U
Unity TechnologiesSoftware Development
Remote, Washington, USAFull-TimeEntry
Salary112,700 - 169,000 USD per year
Apply NowOpens the employer's application page
Job Details
- Languages
- English
- Required Skills
- PythonApache AirflowMachine LearningPyTorchSparkTensorflowDistributed Systems
Requirements
- Bachelor’s degree in Computer Science, Machine Learning, Systems, or a related field
- Strong foundation in machine learning systems, distributed systems, or large-scale data processing
- Experience with Python and working with data-intensive workloads
- Familiarity with ML frameworks (e.g., PyTorch, TensorFlow) and/or distributed systems (e.g., Ray, Spark)
- Experience (academic or applied) with data pipelines, model training workflows, or large datasets
- Strong problem-solving skills and ability to translate research ideas into practical systems
- Interest in building scalable, reliable infrastructure for machine learning
Responsibilities
- Build and maintain data pipelines that generate training datasets for machine learning models and experimentation
- Contribute to infrastructure that supports distributed training workflows (e.g., PyTorch, Ray)
- Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to support multi-stage ML pipelines
- Improve reproducibility and reliability through dataset validation, monitoring, and testing
- Partner with ML engineers to support experimentation and model iteration
- Help optimize performance and efficiency across data processing and training systems
- Contribute to the evolution of our offline ML platform architecture as it scales
View Full Description & ApplyYou'll be redirected to the employer's site
