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 • Fintech • Insurance • Marketing Tech • Software • Analytics
Leads Compensation Operations for monthly commission and compensation processing serving approximately 1,500 agents and managers. Oversees an eight-person team, ensures accurate and timely monthly close, drives process improvement and automation, promotes AI adoption, reconciles transactions across finance systems and carrier statements, partners with compensation design and technology teams, and advises organizational leadership on compensation operations.
Top Skills:
Artificial IntelligenceCompensation Technology
.png)
Consumer Web • eCommerce • Machine Learning • Software • Sports • Analytics
Design and develop scalable cloud-based backend architectures for PSA’s consumer-facing Set Registry platform. Own features through architecture, implementation, testing, deployment, and production release. Build APIs for web and mobile applications, improve databases and existing applications, troubleshoot complex systems, and collaborate with product and engineering teams. Contribute to technical design reviews, documentation, monitoring, security practices, and automated testing.
Top Skills:
Amazon Api GatewayAmazon EcsAmazon EventbridgeAmazon SnsAmazon SqsAws CdkAws CloudformationAws LambdaBatch ProcessingDockerDynamoDBEvent-Driven ArchitecturesKubernetesMicroservicesPostgresPythonRest ApisTerraform
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Develop autonomy evaluation metrics, statistical and machine-learning analyses, dashboards, and pipelines for simulation and on-road autonomous vehicle testing. Analyze perception, prediction, and planning performance; identify anomalous behavior and critical scenarios; apply VLMs and LLMs where appropriate; and support release gating and safety decisions. The role requires production Python, C++ debugging, technical leadership, cross-functional collaboration, and rigorous software engineering practices.
Top Skills:
C++Computational GeometryLarge Language ModelsLinear AlgebraMachine LearningNumpyPandasPythonPyTorchRosScipySQLStatistical ModelingTime-Series AnalysisVision-Language Models
What you need to know about the Austin Tech Scene
Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.
Key Facts About Austin Tech
- Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
- Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
- Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
- Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
- Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center
