Staff Engineer, RL Systems & ML Infrastructure

Goaly; Menlo Park (CA); Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Meals and office benefits

Visa sponsorship

Location-based hybrid policy

Job summary

Goaly is hiring for an ambitious role at the intersection of distributed systems, ML infrastructure, and high-performance computing. You will architect scalable RL pipelines, run environment orchestration, and optimize training and inference for long-running experiments in a hybrid, office-based setting in Menlo Park, CA.

You will collaborate with researchers to identify bottlenecks, redesign critical paths, and deliver robust platform capabilities that scale with research demand while

Qualifications

• A track record building and operating distributed systems, ML infra, or performance-critical backends.

Responsibilities

• Architect and implement end-to-end RL pipelines coordinating rollout generation, environment execution, reward computation, training, evaluation, and checkpoint promotion.

Skills

Distributed systems

ML infrastructure

Python

C++

Rust

Go

Concurrency

Observability

Profiling

Ownership

Tools

PyTorch

JAX

Kubernetes

Ray

Megatron

DeepSpeed

TensorRT-LLM

vLLM

Job description

Goaly is hiring for an ambitious role at the intersection of distributed systems, ML infrastructure, and high-performance computing. You will architect scalable RL pipelines, run environment orchestration, and optimize training and inference for long-running experiments in a hybrid, office-based setting in Menlo Park, CA.

You will collaborate with researchers to identify bottlenecks, redesign critical paths, and deliver robust platform capabilities that scale with research demand while