All jobs → Partner boards

Member of Technical Staff - ML Performance

Veeda AI • CA, US • published 2026-10-06 19:33:47.349262

Description

**Member of Technical Staff \- ML Performance** =============================================== **About Us** ------------ Veeda AI is building the next generation of multimodal foundation world models for Physical AI. We're a small, fast\-moving team of engineers and researchers from leading AI labs, tackling some of the most challenging problems at the intersection of AI, robotics, and embodied intelligence. If you're excited about pushing the boundaries of what's possible with Physical AI, you'll have the opportunity to make an outsized impact from day one. **Responsibilities** -------------------- * **Distributed Training Throughput:** Own step time and model FLOPs utilization for multi\-node video world model training, choosing the tensor, context, and expert parallelism mix in PyTorch FSDP2 and Megatron\-Core rather than inheriting a default. * **Inference Throughput Optimization:** Profile end\-to\-end latency, GPU utilization, memory pressure, and kernel efficiency with PyTorch Profiler, Nsight Systems, Nsight Compute, and torch.utils.benchmark, then improve throughput through torch.compile/Inductor, CUDA Graphs, mixed and low precision, quantization, operator fusion, and mu...

Get jobs like this daily in Telegram: @careeri_bot