ML Performance Engineer — Distributed Training & Kernels
Job in
California, Moniteau County, Missouri, 65018, USA
Listed on 2026-08-27
Listing for:
Veeda
Apprenticeship/Internship
position Listed on 2026-08-27
Job specializations:
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Job Description & How to Apply Below
Veeda AI is seeking a Member of Technical Staff - ML Performance to optimize distributed training for multi-node video world models. You will own step time, FLOPs utilization, memory patterns, and parallelism strategies across NVLink, NCCL, and PyTorch FSDP2 in real workloads.
You’ll implement kernels, tune CUDA/Triton components, improve precision stability, and contribute to fault diagnostics and elastic checkpointing.
#J-18808-LjbffrTo View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
Search for further Jobs Here:
×