Member of Technical Staff - ML Performance

• Bachelor's degree or equivalent hands-on experience in Computer Science, Computer Engineering, or a related technical field.
• Deep hands-on experience with PyTorch and at least one large-scale parallelism stack (FSDP2, Megatron-Core, TorchTitan, or DeepSpeed) on real multi-node jobs, not single-node approximations.
• Fluency in Python and C++/CUDA with the ability to predict where a kernel will stall from its memory access pattern before profiling.
• Experience profiling live training and inference runs with Nsight Systems or the PyTorch profiler and translating traces into quantifiable step-time or MFU improvements.
• Expertise in at least one of low-precision numerics, kernel authoring, or large-run fault diagnosis, and credibility in the others.

• Experience optimizing training or inference workloads across very large GPU clusters, including topology-aware placement, scaling efficiency, performance isolation, and diagnosing failures that emerge only at fleet scale.
• Experience writing Triton, CUTLASS, or CuTe-DSL kernels, or contributing to open-source kernel libraries.
• Experience implementing context or sequence parallelism for long-horizon video or high-token-count models.
• Experience running or porting large training workloads on AMD GPUs (ROCm) or Google TPUs (JAX/XLA).
• Experience building fault-tolerant training with elastic world size, dynamic node re-queueing, or asynchronous distributed checkpointing.
• Experience optimizing generative inference for interactive rollouts, including few-step samplers, distillation, and KV or latent caching.
• Publications or presentations on machine learning systems, compilers, or high-performance kernels.

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.