1 day ago
Washington, DC, USA or Mountain View, CA, USAIntern
H1B sponsor
Base Salary
$92k - $152k/yr
Responsibilities
- Build and maintain data pipelines that generate training datasets for machine learning models and experimentation.
- Contribute to distributed training infrastructure using tools such as PyTorch and Ray.
- Use workflow orchestration systems such as Airflow or Flyte for multi-stage ML pipelines.
- Improve reproducibility and reliability through dataset validation, monitoring, and testing.
- Partner with ML engineers on experimentation and model iteration.
- Optimize performance and efficiency across data processing and training systems.
- Help evolve 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 data-intensive workloads.
- Familiarity with ML frameworks such as PyTorch or TensorFlow and/or distributed systems such as Ray or Spark.
- Academic or applied experience with data pipelines, model training workflows, or large datasets.
- Strong problem-solving skills and ability to translate research ideas into practical systems.
- Interest in scalable and reliable machine learning infrastructure.
- Preferred: experience with Airflow or Flyte, exposure to large-scale data platforms, or publications/research in ML systems or distributed systems.
Benefits
- Comprehensive health, life, and disability insurance, with eligibility varying by country and employment status.
- Commute subsidy, employee stock ownership, retirement or pension plans, vacation and personal days, and new-parent and family-care programs.
- Office food snacks, mental health and wellbeing programs, employee resource groups, and a global employee assistance program.
- Training and development programs, volunteering opportunities, and donation matching.
- The position is offered in geographic salary zones, with the final base offer depending on location, experience, background, and skills.
Tech Stack
Categories
Data EngineeringML Engineering
About Unity
Unity is a public software company that builds the Unity real-time 3D engine and services for creating, operating, and monetizing games and interactive 3D content across mobile, PC, console, and XR. It sells subscription licenses and usage-based services such as cloud build, ads, and analytics to studios and enterprises in gaming, automotive, film, and AR/VR. Founded in 2004 and headquartered in San Francisco, it trades on the NYSE (U).