Remote ML Platform Engineer - New Grad

United States Digital Space LLC; United States; Remote; USD 98,000 - 139,000; Full time

Benefits offered by this job

Medical/dental/vision coverage

Equity refreshers

Remote work reimbursement

PTO

Employee assistance programs

and more

Job summary

the company is seeking a New Grad Software Engineer to join the ML platform and ranking infrastructure team. You will work at the intersection of Machine Learning, Distributed Systems, and GPU Serving, shipping to production early in the role with mentorship and guidance.

You will help build scalable ML infrastructure, support production models, and contribute to PyTorch-first standardization while improving developer velocity and feature delivery.

Qualifications

• Availability for meetings during the company’s coordination hours (Mon-Fri: 9am-3pm Pacific Time).
• Graduate in 2025 or 2026 with a B.S., M.S., or Ph.D. in Computer Science, Engineering or a related field.
• Genuine interest in large-scale distributed systems, infrastructure, and machine learning.
• Knowledge of Python, Go or C++, or the ability to learn them quickly.
• A passion for learning and always improving yourself and the team around you.

Responsibilities

• Help build and maintain the core infrastructure that powers the company's ML platform, ensuring high availability, scalability, and performance.
• Build and improve the distributed systems that serve our ML models in production, from Large Recommendation Models to Large Language Models.
• Work on GPU model serving, optimizing latency, throughput, and cost to support larger and more capable models.
• Contribute to platform initiatives such as PyTorch-first standardization and ML ecosystem modernization.
• Improve ML developer velocity by building tooling that helps ML engineers develop, test, and deploy models more efficiently.
• Modernize our feature store so ML engineers can get new features into production faster.
• Participate in the team's on-call rotation, helping resolve production issues as you grow your knowledge and ownership of the platform.

Skills

Python

Go

C++

Learning agility

Education

Bachelor's degree in Computer Science or related field

2025 or 2026 graduate

Tools

Kubernetes

Docker

AWS

Job description

the company is seeking a New Grad Software Engineer to join the ML platform and ranking infrastructure team. You will work at the intersection of Machine Learning, Distributed Systems, and GPU Serving, shipping to production early in the role with mentorship and guidance.

You will help build scalable ML infrastructure, support production models, and contribute to PyTorch-first standardization while improving developer velocity and feature delivery.