Senior ML Platform Engineer – Distributed GPU & Game AI

WorkGenius Group; Santa Monica (CA); On-site; USD 124,000 - 207,000; Full time

Job summary

WorkGenius Group in Santa Monica, CA seeks a Senior Machine Learning Engineer to build and scale distributed ML training systems across multi-node GPU clusters and to integrate platform components for games.

You will develop automated pipelines, inference serving optimized for latency and cost, and create tools to accelerate the ML lifecycle, while collaborating with game teams and mentoring engineers.

Qualifications

• Bachelor’s degree in Computer Science or related field or equivalent experience with 3+ years in ML/platform/systems engineering.
• Experience operating distributed systems and production GPU workloads including distributed ML or agent training with RL/IL.

Responsibilities

• Build and operate distributed ML training systems across multi-node GPU clusters.
• Develop automated ML pipelines with data/model versioning, validation, and experiment tracking.
• Build and optimize inference-serving systems for latency, cost, and reliability.
• Create developer tools, SDKs, and templates to streamline ML lifecycle.
• Monitor system health, troubleshoot production issues, and mentor engineers.

Skills

Distributed ML

GPU Clusters

Reinforcement Learning

Imitation Learning

PyTorch

Python

C++

Ray/RLlib

Inference Serving

ML Platform Engineering

Education

Bachelor’s degree in Computer Science or related field

Tools

PyTorch

Ray

RLlib

C++

Python

Multi-node GPU orchestration

Job description

WorkGenius Group in Santa Monica, CA seeks a Senior Machine Learning Engineer to build and scale distributed ML training systems across multi-node GPU clusters and to integrate platform components for games.

You will develop automated pipelines, inference serving optimized for latency and cost, and create tools to accelerate the ML lifecycle, while collaborating with game teams and mentoring engineers.