DeveloperJobs.io
← Back to all jobs
Unity Technologies SF

Machine Learning Engineer

Unity Technologies SF
Mountain View, CA $104k - $152k/yr Full time Posted 1d ago
Apache Airflow Artificial Intelligence Automation Big Data Bigdata Data Integration Data Pipeline Data Platform Data Processing Deep Learning Distributed Training ETL Machine Learning Machine Learning Engineer Ml Ops PyTorch TensorFlow
Apply Now

Job Description

Machine Learning Engineer on Unity's Offline Infrastructure team in Mountain View, CA (onsite), building data pipelines for training data generation and distributed model training to scale ML workflows.

Responsibilities

  • Design and maintain data pipelines that generate training datasets for ML models and experiments
  • Contribute to infrastructure that supports distributed training workflows using tools like PyTorch and Ray
  • Collaborate with workflow orchestration tools such as Airflow, Flyte, or similar to enable multi-stage ML pipelines
  • Improve reproducibility and reliability via dataset validation, monitoring, and testing
  • Partner with ML engineers to support experimentation and model iteration
  • Optimize performance and efficiency across data processing and training pipelines
  • Contribute to the evolution of the offline ML platform architecture as it scales

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 (through research or projects)
  • 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

Technologies

  • Python
  • PyTorch
  • TensorFlow
  • Ray
  • Spark
  • Airflow
  • Flyte

Compensation

Salary: USD 104,100 - 152,000 per year

Benefits

  • Comprehensive health, life, and disability insurance
  • Commute subsidy
  • Employee stock ownership
  • Competitive retirement/pension plans
  • Generous vacation and personal days
  • Support for new parents through leave and family-care programs
  • Office food snacks
  • Mental Health and Wellbeing programs and support
  • Employee Resource Groups
  • Global Employee Assistance Program
  • Training and development programs
  • Volunteering and donation matching program

Additional information

  • Relocation support is not available for this position
  • Work visa/immigration sponsorship is not available for this position

Similar Jobs