Machine Learning Engineer, Offline Infrastructure (Entry-Level / New Grad PhD)
Mountain View, CA, United States
Posted 4 hours ago
Last seen 4 hours ago
Active
full-time
AI & Machine Learning
$112,700—$169,100 USD
Job Summary
Join Unity's AI & Machine Learning team as an Entry-Level Machine Learning Engineer to build scalable offline ML infrastructure, applying research-driven thinking to real-world problems and powering AI-driven decision-making across the company.
About this job
Unity Vector is building an offline ML platform for insights, experimentation, attribution, and AI-driven decision-making. This role involves building and evolving infrastructure for training data generation, ML workflows, and distributed model training. You will contribute to scalable and efficient ML pipelines, bridging research and production.
Requirements
- PhD 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
- Experience with workflow orchestration systems (Airflow, Flyte, etc.) (Nice to Have)
- Exposure to large-scale data platforms (data lakes, warehouses, streaming systems) (Nice to Have)
- Publications or research in ML systems, distributed systems, or related areas (Nice to Have)
Benefits & Perks
- 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
Apply for this position
You'll be redirected to the company's application page to complete your application.
Related Jobs
Similar positions you might be interested in
Explore Categories
Browse jobs by department and skill area