ML Performance Engineer: Scale GPU-Driven Training
Decisive Point
Applied Intuition, Inc. is a Silicon Valley leader powering the future of physical AI. We seek a Performance Engineer to optimize large-scale ML workloads, focusing on distributed training, batch inference, and cost-effective data processing.
You will own profiling across the stack, identify where compute and wall-clock time are spent, and drive improvements that speed time-to-result and reduce processing costs. Join a team that values collaboration and technical excellence.
#J-18808-Ljbffr$189.3k - $290.7k
...transportation on a global scale. Role:Are you... ...scenarios.As a Staff ML Infra Engineer, you will drive the development... ...dataset generation, training, evaluation, and... ...training pipelines that are performant, easy to use, and... ...training across large GPU/CPU clusters or other...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$213k - $263k
...states. The Waymo ML Infrastructure... ...different modeling teams. Scale and efficiency are... ...are looking for engineers with ML software &... ...custom NVIDIA GPU kernel development... ...Analyze ML workload performance at the hardware... ...experience with designing, training and debugging deep...TrainingPerformanceFull timeRemote work$155.42k - $395.9k
...DescriptionAbout the Team:The ML Compute... ...building AI-driven products for GM... ...supports the training and deployment... ...a focus on performance, availability,... ...to maximizing GPU utilization across... ...Senior Software Engineer to join our team and help us scale our platform for...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours- ...Machine Learning Engineer to join our Models... ...challenge of distributed training of large models on... ...generative AI at scale.THE PERSON:The... ...training pipeline performance.Optimize the... ...EXPERIENCE:Experience with ML/DL frameworks such... ....Experience with GPU kernel...TrainingPerformance
$189k - $300k
...transportation on a global scale. The Data Scaling team... ...works on and delivers ML models to the product... ...directly impacting AV product performance through smart use of... ...foundation model pre-training and fine-tuning with... ...high-impact team of AI/ML engineers, data scientists and...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$153.2k - $234.1k
...transportation on a global scale.Role Overview:Are you... ...every machine learning engineer working on our cutting-... ...the safety and performance of the car, rather than... ...driverless vehicles.As a Senior ML Infra Engineer, you... ...machine learning model training and evaluation workflows...TrainingPerformanceFull timeWork at officeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours- ...Applied Intuition, Inc. is seeking a performance engineer to accelerate large-scale ML workloads in the data center. You will own profiling, optimization, and cost-efficiency for distributed training and large offline inferences. You will work across accelerators, ML...TrainingPerformance
- ...Applied Intuition in Sunnyvale is seeking a Performance Engineer to accelerate large-scale ML workloads in data centers. You will optimize distributed training across many nodes and improve batch inference on petabyte-scale sensor logs, targeting throughput and cost-per...TrainingPerformance
- ...Applied Intuition, Inc. in Sunnyvale, CA, is seeking a performance engineer to optimize large-scale ML workloads in the datacenter. This role focuses on distributed training across many nodes and high-throughput batch inference over petabytes of real-world autonomy...TrainingPerformance
- Nebius B.V. is building an AI training and model post-training... ...the infrastructure for large-scale training, RL experiments,... ...intersection of distributed systems, GPU performance, and ML framework integration. The... ...strong Python and PyTorch engineering skills, hands-on experience...TrainingPerformance
- Rhoda AI in Mountain View is seeking a Staff / Principal ML Training Systems Engineer to lead the performance of large-scale multimodal training systems. This role involves improving training efficiency and collaborating closely with research teams to accelerate model iteration...TrainingPerformance
$150k
...to a knowledge-driven economy. As... ...foundation model training, alongside... ...scientists, and engineers, tackling the most... ...global hub for high-performance computing in... ...Distributed ML Engineer will play... ..., and large-scale machine learning... ...solutions to enhance GPU utilization...TrainingPerformanceFull timeWork experience placementVisa sponsorship$250k - $350k
.../Staff level Inference Engineers to accelerate the performance of Pika's AI-driven products. In this highly... ...acceleration, GPU parallelism, advanced model... ...leading user experiences at scale.You will design and... ...into production.Improve Training Efficiency: (Bonus) Contribute...TrainingPerformanceWork at office3 days per week- ...research, software engineering, hardware... ...machine learning at scale on Apple... ...acceleration, building ML compilers and... ..., competitive performance on Apple... ...models with strong, driven performance across... ...architectures, training vs. inference... ...Swift, and/or GPU programming paradigms...TrainingPerformance
$300k
...to a knowledge-driven economy. As... ...foundation model training, alongside... ...scientists, and engineers, tackling the most... ...global hub for high-performance computing in... ...a distributed ML infrastructure... ...help extend and scale our training... ...multi-node, multi-GPU clusters •...TrainingPerformanceFull timeFlexible hours$224k - $356.5k
...Machine Learning Engineers to join our... ...pioneers of the GPU—the visual cortex... ...our AI agents are trained on the most diverse... ...fine-tune large-scale models,... ...full lifecycle of ML software, including performance optimization, testing... ...gap between data-driven approaches and...TrainingPerformanceFull time$119.25k - $150.85k
...hands-free miles driven—demonstrating... ...unprecedented scale. Within GM AV,... ...learning models from training frameworks (e.g... ...to build the ML deployment... ...and we’re hiring engineers to help deliver... ...deployment workflows, performance investigations,... ...ML compilers, GPU programming (...TrainingPerformanceFull timeInternshipLocal areaWork from homeRelocation packageFlexible hours$296.3k
...seeking a Principal AI Engineer to lead the design... ...that powers large-scale training and cloud... ...generation of AI-driven driving systems. We... ...optimize core AI/ML platform infrastructure... ...scalability, and performance across the AI/ML platform... ...systems, GPU computing, and cloud...TrainingPerformanceFull timeLocal areaRemote workWork from homeFlexible hours$189.4k - $300.6k
...transportation on a global scale. Are you passionate... ...development. We engineer high-performance tools that identify top... ...with data-intensive ML teams to drive rapid... ...drive actionable, data-driven decisions with... ...back to data selection, training, and launch decisions...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$170k - $240k
...oriented, impact delivering-driven expert in ML Training Infrastructure with a... ..., reliable, and high-performance AI/ML platform... ...initiatives. As a Senior ML Engineer, you will collaborate... ...model training at scale.Model training... ...distributed computing, GPU computing, and cloud...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours- ...OverviewAs our Staff Software Engineer, ML infra Engineer for... ...data needed to train complex ML models and... ...growth strategies are driven or boosted by ranking... ...pipelinesDevelop and scale data infrastructure that... ...structures, algorithms, performance complexity, and implications...TrainingPerformanceTemporary work
- ...You will contribute across pre-training to post-training, designing... ...training to push state-of-the-art performance. You will work with cutting... .... The role emphasizes strong ML/DL fundamentals, proficiency in Python, and a research-driven mindset, with opportunities to...TrainingPerformance
$90.1k - $191.8k
...world!The Data Labeling Engineering team designs, builds,... ...engineering, and ML, defining labeling strategies... ...that create reliable training data at scale.Our team builds the... ...test scalable, high‑performance user experiences and... ...and integrate ML‑driven data annotation (pre‑...TrainingPerformanceFull timeWork experience placementLocal areaRemote workWork from homeRelocation packageFlexible hours$174.9k - $261.3k
...world!The Data Labeling Engineering team designs, builds,... ...engineering, and AI/ML, defining the... ...that create reliable training data at scale. Our tools and platform... ...test scalable, high‑performance user experiences and... ...design and integrate ML‑driven data annotation (pre‑...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours- ...AI is seeking a Machine Learning Engineer in Palo Alto, California, to build... .... You will play a key role in scaling our Ray and PyTorch-based systems... ...directly influence the efficiency and performance of our data processing and model training pipelines. #J-18808-Ljbffr...TrainingPerformance
$165.2k - $223.6k
...Software Development Engineer to own the design... ..., developing high-performance compute kernels,... ...operate at frontier scale with large... ...kernels for a custom ML accelerator architecture... ...across CPU, GPU, simulator, and hardware... ...architecture, training/inference lifecycles...TrainingPerformanceLocal areaFlexible hours- ...Waymo is hiring engineers to build ML systems that train and improve pre-trained models for the Waymo Driver, tackling... ...learning at unprecedented scales. You will collaborate with researchers... ...to deploy RL techniques across CPU/GPU/TPU, design scalable training pipelines...Training
$296.3k - $423.9k
...transportation on a global scale. We are looking for a... ...shape the algorithms and ML systems that translate sensor... .... You will lead a high-performing team of engineers building ML-driven trajectory generation... ...environments. Architect scalable training pipelines using large-...TrainingPerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours- ...The Role As an ML Engineer within the Application Engineering... ...in terms of core driving performance and feature-level... ...’ll design and deliver ML-driven behaviors that scale from assisted to autonomous... ...Influence architecture choices, training methodologies, and...TrainingPerformanceFull timeWork at officeWork from home
$150k
...contributions to a knowledge-driven economy. As... ...foundation model training, alongside world-... ...scientists, and engineers, tackling the most... ...hub for high-performance computing in deep... ...performance of large-scale machine learning models... ...Machine Learning (ML) models, ML...TrainingPerformanceFull timeWorldwideVisa sponsorship
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to ML Performance Engineer: Scale GPU-Driven Training. Be the first to apply!
- ai ml engineer Sunnyvale, CA
- senior ml engineer Sunnyvale, CA
- computer vision machine learning engineer Sunnyvale, CA
- machine learning engineer Sunnyvale, CA
- machine learning ai engineer Sunnyvale, CA
- machine learning software engineer Sunnyvale, CA
- senior performance tester Sunnyvale, CA
- lead performance test engineer Sunnyvale, CA
- performance windows Sunnyvale, CA
- performance specialist Sunnyvale, CA


