Skip to content
CV Rocket
Get started
How it worksThe enginePricingBlogContactSign in

Open nowPosted 12 days ago

ML Platform Engineer, Distributed Training

Egen23 open roles

Pay
$159,300 – $219,245 a year
Where
Remote
Work mode
Remote
Get the CV for this job

From $25 per CV, paid once. No subscription.

Your applicationOpen nowML Platform Engineer, Distributed TrainingEgen · Remote

  1. YouYes, apply to this one.

  2. CV RocketCV written for this posting.

  3. 25 readersRecruiter, hiring manager, skeptic. Round after round.

  4. CV RocketApplied on Egen's own form.

The reply lands in your private mailbox

3×more interviews than doing it yourself with ChatGPT.

The clock on this job

Early applications get read.

12.9% of postings close within 7 days. Measured by our own scanner across the market. Egen postings stay open a median of 23 days.

Share of postings closed within
  1. 2.2%1 day
  2. 4.7%3 days
  3. 12.9%7 days
  4. 21.0%14 days
  5. 35.6%30 days
  6. 58.5%60 days
This job: posted 12 days ago

Egen median: 23 days open

The posting

About Egen:

Egen is a fast-growing and entrepreneurial company with a data-first mindset. We bring together the best engineering talent working with the most advanced technology platforms, including Google Cloud and Salesforce, to help clients drive action and impact through data and insights. We are committed to being a place where the best people choose to work so they can apply their engineering and technology expertise to envision what is next for how data and platforms can change the world for the better. We are dedicated to learning, thrive on solving tough problems, and continually innovate to achieve fast, effective results. If this describes you, we want you on our team.

Want to learn more about life at Egen? Check out these resources in addition to the job description.

Meet Egen

Life at Egen

Culture and Values at Egen

Career Development at Egen

Benefits at Egen

About the opportunity

We're looking for an engineer who is equally at home running a GPU training platform and the large-scale model training that runs on it. You'll be embedded with a healthcare and life-sciences AI client that trains foundation models on a large GKE-based H200 and GB200 cluster. The client is reorganizing so that its foundation-model platform and its agent platform teams work more closely together.

The role centres on the training stack and the scientists who use it. Solid infrastructure knowledge and experience are required, but the client already has strong SRE and infrastructure capacity, so running the cluster isn't the focus: you'll partner with that team and understand the training workloads well enough to tune, benchmark, and debug them end to end. You'll work alongside other Egen engineers and client scientists, and you'll own your area outright.

What You Will Do:

  • Work with the client's infrastructure team on the GKE training platform (GPU node pools, job queueing and scheduling, quota and capacity) from the training workload's side.
  • Diagnose and fix platform failures at the root (for example a controller that keeps running out of memory) rather than restarting them.
  • Tune multi-node GPU networking (RDMA/GPUDirect, NCCL) for training throughput.
  • Run, scale and debug multi-node distributed training with PyTorch (FSDP, DeepSpeed, Megatron or NeMo-style) and Ray Train.
  • Profile jobs and remove bottlenecks across compute, memory, input pipelines and interconnect.
  • Design and run performance benchmarks for the cluster and training stack: throughput, scaling efficiency, cost per run.
  • Build and maintain model evaluation harnesses so changes to the platform or training recipes are measured, not assumed.
  • Instrument training and the platform with telemetry and monitoring, and keep end-to-end pipeline tests running against what users actually run.
  • Treat the platform as your own: proactively find issues, plug gaps and propose solutions before anyone asks, keep the team informed of status and risks, and look after the scientists who use it.
  • Explain and defend your design decisions to client engineers and scientists in detail.
  • Stay effective while the client's priorities and team structure change.

Your Technical Toolkit:

  • Core Languages: Mastery of Python and shell scripting
  • AI/LLM Ecosystem: Extensive experience with Google Gemini, GPT-4, or LLaMA; deep knowledge of Prompt Engineering and Fine-tuning
  • Data & Search: Expertise in Vector Databases (Vertex AI Vector Search, pgvector, etc.) and implementing Semantic Search
  • Infrastructure: Hands-on experience with Google Cloud (Vertex AI) and building scalable software architectures
  • Frameworks: Proficiency in LangChain, LlamaIndex, or similar orchestration layers
  • Mindset: A strong software engineering foundation—you write clean, maintainable code and understand the full SDLC

Basic Qualifications:

  • 5+ years of engineering experience, with recent hands-on work on GPU training infrastructure and large-scale training.
  • Strong Python software engineering: production-quality code, tests and code review. Coding is central to this role, not an afterthought.
  • GKE and GPU scheduling: Kubernetes on Google Cloud with GPU node pools, job queueing (Kueue or similar), and capacity and quota management.
  • Distributed PyTorch and Ray Train: has personally run and debugged multi-node training jobs.
  • RDMA and GPU networking: GPUDirect/RDMA, NCCL tuning, multi-node GB200 (or H200) topology.
  • Benchmarking and evals: has designed performance benchmarks and model evaluations, and used them to make decisions.
  • Deep enough understanding to explain how and why a system works, without leaning on AI coding tools to carry the reasoning.
  • A track record of visible ownership: present, communicative, and accountable on a demanding team.

Personal Attributes:

  • Healthcare or life-sciences data (genomics, clinical, biomedical), and working under HIPAA.
  • Observability and telemetry for training platforms: metrics, tracing, alerting, end-to-end pipeline testing.

Compensation & Benefits:

This role is eligible for our competitive salary and comprehensive benefits package to support your well-being:

- Comprehensive Health Insurance

- Paid Leave (Vacation/PTO)

- Paid Holidays

- Sick Leave

- Parental Leave

- Bereavement Leave

- 401 (k) Employer Match

- Employee Referral Bonuses

Check out our complete list of benefits here - >https://egen.ai/people/#benefits

Important: All roles are subject to standard hiring verification practices, which may include background checks, employment verification, and other relevant checks.

EEO and Accommodations:

Egen is an equal opportunity employer and is committed to inclusion, diversity, and equity in the workplace. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, national origin, disability status, protected veterans’ status, or any other characteristic protected by federal, state, or local laws. Egen will also consider qualified applications with criminal histories, consistent with legal requirements. Egen welcomes and encourages applications from individuals with disabilities. Reasonable accommodations are available for candidates during all aspects of the selection process. Please advise the talent acquisition team if you require accommodations during the interview process.

From $25, paid onceGet the CV for this job

What happens when you press

One press. We do the rest.

  1. A CV for this posting

    Written against Egen's own wording, from every piece of relevant proof in your profile.

  2. 25 readers review it

    Recruiter, hiring manager, skeptic and more read every draft, round after round. You get the best round.

    The review screen in CV Rocket: how each CV was read, round by round.

  3. We apply on Egen's form

    Our application engine gets through the hardest forms there are. Where a question needs you, AI suggests the best answer. Don't want us applying from our IP addresses? Use our Chrome extension: we apply straight from your own browser.

    An application in CV Rocket: every answer filled in on the employer's form.

  4. Every reply, sorted

    Egen's answer lands in your private mailbox, and we classify it on arrival: interview, question, rejection.

    The CV Rocket inbox: each employer reply classified as an interview, an action or a rejection.

  5. Reply with AI

    AI helps you write the email, checks it and sends it. We show you whether the recruiter read it.

  6. The interview in your calendar

    Full integration with your calendar. The invitation goes straight in.

    An interview invitation in the CV Rocket inbox, added to the candidate's calendar.
Get the CV for this job

From $25 per CV, paid once. No subscription.

Why it works

3×

more interviews than doing it yourself with ChatGPT.

ChatGPT writes a CV and never learns what happened to it. We see every reply. For each CV we know:

  • How it was written, and how the review scored it
  • When we applied, and how long after the posting went up
  • Which posting, which company, which city
  • Who got the interview, and who heard nothing

That is how we know which CVs get called.

Get the CV for this job

From $25 per CV, paid once. No subscription.

The numbers game

More applications. More interviews.

Every application goes out with its own CV, written for that posting and paid once. Send enough of them and the law of large numbers finds you the job.

Get the CV for this job
By hand5–10
With CV Rocket100
applications a day

Nearby

Live postings like this one

Same employer first, then the same role elsewhere.

Before you press

Straight answers

Get the CV for this job

From $25 per CV, paid once. No subscription.

What if my background isn't good enough?

We make the most of the background you have. The CV uses every piece of relevant proof your profile holds, and one of the 25 readers reads your whole profile and flags what the CV left out.

Do you really apply for me?

Yes, on the employer's own form, the hardest ones included. Where a question needs you, you answer it right there and AI suggests the best answer. Don't want us applying from our IP addresses? Use our Chrome extension: we apply straight from your own browser.

Is it a subscription?

No. You pay once per CV, from $25. Every application goes out with its own CV, written for that posting.

One job. One CV.
Paid once.

Pick the posting you want. We write for it, apply for you and catch the reply.

Get the CV for this job

From $25 per CV, paid once. No subscription.