EngineerJobs.io
← Back to all jobs
Unity Technologies SF

Machine Learning Engineer

Unity Technologies SF
Mountain View, CA $104k - $152k/yr Full time Posted 9d 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 in Mountain View, CA (onsite) on Unity's Offline Infrastructure team, building data pipelines for training data generation and distributed model training to scale ML workflows.

Salary

USD 104,100 - 152,000 per year

Responsibilities

  • Design and maintain data pipelines that generate training datasets for ML models and experiments
  • Contribute to infrastructure supporting distributed training workflows using PyTorch and Ray
  • Collaborate with workflow orchestration tools like Airflow or Flyte to support multi-stage ML pipelines
  • Enhance reproducibility and reliability through dataset validation, monitoring, and testing
  • Partner with ML engineers to back experimentation and model iteration
  • Optimize performance and efficiency across data processing and training systems
  • 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 ML systems, distributed systems, or large-scale data processing demonstrated through research or projects
  • Experience with Python and working with data-intensive workloads
  • Familiarity with ML frameworks (PyTorch, TensorFlow) and/or distributed systems (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

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 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 assistance is not available for this position
  • Work visa/immigration sponsorship is not available for this position

Similar Jobs