Member of Technical Staff, Infrastructure

Added
3 hours ago
Type
Full time
Salary
Upgrade to Premium to se...

Related skills

python pytorch distributed systems jax ml infrastructure

📋 Description

  • Scale our post-training stack, supporting SFT and RL on leading open-weight models.
  • Build distributed rollout infrastructure that coordinates model inference, tool execution, and
  • Build the infrastructure for reproducible agent environments where models can interact with
  • Own the performance and reliability of training runs, including GPU utilization, checkpointing
  • Identify and remove bottlenecks across training and inference to shorten the path from research
  • Build the internal tools and services the team runs on: training data management and versioning

🎯 Requirements

  • Experience building ML infrastructure, distributed systems, or research tooling, with ownership of
  • Strong Python and generalist software engineering skills across backend services, data systems, and
  • Experience with distributed training or inference, including diagnosing performance bottlenecks and
  • Understanding of post-training workflows; experience with RL rollouts, agent environments, or
  • Practical judgment about what to build versus adopt, and the ability to partner with researchers

🎁 Benefits

  • Salary: $230,000 – $340,000 USD
  • Equity
  • Bonus
Get alerts for new remote engineering jobs
A weekly email with new matches. Unsubscribe anytime.
Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Related Engineering Jobs

See more Engineering jobs →