Operate and optimize large-scale LLM pre-training on 1,000+ GPU clusters using PyTorch, DeepSpeed, or Megatron-LM. Improve networking (InfiniBand/RDMA), memory management, checkpointing, and failure recovery. Manage SLURM/Kubernetes GPU clusters and apply systems engineering (C++, CUDA, Python) and 3D parallelism techniques.
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).
Similar Jobs
Artificial Intelligence • Cloud • Consumer Web • Productivity • Software • App development • Data Privacy
Own the full B2B SaaS sales cycle, from outbound pipeline generation through closing and renewal. Build strategic relationships with mid-market and enterprise customers, lead discovery, apply MEDDICC or SPICED qualification, communicate business value, forecast accurately, and consistently achieve revenue targets. Collaborate with internal teams to refine go-to-market strategy, identify new use cases, and influence product and sales initiatives.
Top Skills:
DropboxSalesforce
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Manage an enterprise implementation consulting team delivering complex, multi-site Samsara deployments involving hardware, software configuration, and API integrations. Coach team members, oversee KPIs and OKRs, manage executive stakeholder relationships and escalations, improve implementation processes through automation and data, partner with Sales, Support, Solutions Engineering, and Customer Success, and hire and develop a high-performing team.
Top Skills:
APIsGainsightHardware InstallationIotPsa ToolsSaaS
Cloud • Security • Software • Cybersecurity • Automation
Leads more than 10 engineers across Agent Tools, Agent Observability, and Runner Execution. Responsibilities include hiring, coaching, prioritization, technical design, AI tool standards, safe agent execution, sandboxing, observability, and cross-functional partnership across GitLab’s AI organization.
Top Skills:
Agent HarnessesAi Coding ToolsContainer IsolationGoInfrastructure SecurityModel Context Protocol (Mcp)Observability SystemsPythonReliability EngineeringRubySandboxingTypescript
What you need to know about the NYC Tech Scene
As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.
Key Facts About NYC Tech
- Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
- Key Industries: Artificial intelligence, Fintech
- Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
- Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory
