LLM Pre-training & Distributed Engineer (AI Infrastructure)

Reposted One Month Ago
Boston, MA, USA
In-Office
Mid level
Agency • Artificial Intelligence • Blockchain • Web3
The Role
Design, orchestrate, and optimize large-scale LLM pre-training runs across 1,000+ GPUs using PyTorch/DeepSpeed/Megatron-LM. Improve networking (InfiniBand/RDMA) and memory management, implement 3D parallelism, and automate checkpointing and failure recovery for month-long training jobs on SLURM or Kubernetes GPU clusters.
Summary Generated by Built In

We are seeking a highly skilled LLM Pre-training & Distributed Systems Engineer. This role is essential for orchestrating large-scale machine learning training runs and optimizing  distributed infrastructure. The ideal candidate will have a deep understanding of GPU clusters and extensive experience in system engineering to ensure efficient and reliable training processes.

Responsibilities:

  • Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM.
  • Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors.
  • Automate checkpointing and failure recovery during month-long training runs.

Required Skills:

  • Deep expertise in 3D parallelism (Data, Tensor, Pipeline).
  • Experience managing SLURM or Kubernetes-based GPU clusters.
  • Strong systems engineering background (C++, CUDA, Python).

Skills Required

  • Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM
  • Deep expertise in 3D parallelism (Data, Tensor, Pipeline)
  • Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors
  • Automate checkpointing and failure recovery for month-long training runs
  • Experience managing SLURM or Kubernetes-based GPU clusters
  • Strong systems engineering background including C++, CUDA, and Python
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
7 Employees
Year Founded: 2024

What We Do

Hyphen Connect is a Web3 and AI talent agency and crypto-integrated software solutions provider that connects blockchain, DeFi, NFT, and AI companies with specialized technical and go-to-market talent globally and remotely. They deliver headhunting, data-driven research, and recruitment services across infrastructure, exchanges, gaming, and DeFi projects, plus industry analysis and hiring insights to help clients build engineering and product teams.

Similar Jobs

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Inside Sales Representative

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Remote or Hybrid
10 Locations
40000 Employees
45K-85K Annually

PwC Logo PwC

Deals - Financial Due Diligence Intern - Summer 2028

Artificial Intelligence • Professional Services • Business Intelligence • Consulting • Cybersecurity • Generative AI
Hybrid
8 Locations
370000 Employees
29K-48K Hourly

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Senior Claims Representative

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Hybrid
6 Locations
40000 Employees
46K-84K Annually

SharkNinja Logo SharkNinja

Product Engineer

Beauty • Robotics • Design • Appliances • Manufacturing
In-Office
Needham, MA, USA
4000 Employees
66K-95K Annually

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Artificial Intelligence • Fintech • Software
New York, New York
9 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account