Design, orchestrate, and optimize large-scale LLM pre-training across 1,000+ GPUs. Implement 3D parallelism, manage GPU clusters (SLURM/Kubernetes), optimize InfiniBand/RDMA networking and memory, and automate checkpointing and failure recovery for long training runs.
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
.png)
eCommerce • Fintech • Hardware • Payments • Software • Financial Services
Own the architecture and technical strategy for Square’s go-to-market systems. Build and operate production AI agents, automated workflows, and integrations that improve sales productivity. Lead complex initiatives through architecture, implementation, testing, deployment, and operations while establishing AI evaluation, observability, security, reliability, and human-oversight practices. Partner with Sales, Product, Data, and Engineering leaders to translate business needs into scalable solutions, make build-versus-buy decisions, mentor engineers, and improve business impact through data and user feedback.
Top Skills:
Ai AgentsCrm PlatformsGoLarge Language Models (Llms)RetrievalTool CallingWorkflow Orchestration
eCommerce • Fintech • Hardware • Payments • Software • Financial Services
Own the architecture and technical strategy for Square’s go-to-market technology stack. Build and operate production AI agents, automated workflows, and integrations that improve sales productivity. Lead complex technical initiatives across teams, establish reliable and secure AI evaluation practices, make build-versus-buy decisions, and partner with Sales, Product, Data, and Engineering leaders. Mentor engineers, guide design and implementation, and use data and feedback to improve system adoption and business impact.
Top Skills:
Ai AgentsCustomer Relationship Management (Crm) SystemsGoLarge Language Models (Llms)Retrieval SystemsTool CallingWorkflow Orchestration
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Manage customer onboarding, training, adoption, retention, and growth for the Falcon Shield SaaS security platform. Build trusted relationships with security teams, develop measurable success plans, conduct strategic reviews, demonstrate product value, and advise customers on SaaS security posture and architecture. Partner with Product and R&D teams to communicate customer needs, identify product gaps, and support feature improvements.
Top Skills:
Ai AgentsFalcon ShieldIdentity And Access Management (Iam)Large Language Models (Llms)Saas Security Posture Management (Sspm)Saas Security Tooling
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
