Technical Staff Member for Physical AI

4 hours ago

Toronto, Ontario, Canada Veeda AI Full-time
Contribute to groundbreaking Physical AI innovations at Veeda AI as a Technical Staff Member. Engage in multi-node video world model training while tackling key performance challenges. Veeda AI is seeking a Technical Staff Member to enhance its world-class team working with multimodal foundation models. This role requires deep expertise in PyTorch and parallelism stacks like FSDP2 or Megatron-Core. You will optimize distributed training, inference throughput, and ensure numerical stability for cutting-edge AI applications.

Key Responsibilities
• Own model FLOPs utilization during video world model training
• Profile latency and GPU utilization for improved throughput
• Enhance precision and stability for video tokenizer and VAE layers
• Write and tune custom CUDA and Triton kernels
• Develop detection layer for silent data corruption issues

Requirements:
• Bachelor's degree or equivalent in a technical field
• Hands-on experience with PyTorch in multi-node jobs
• Fluency in Python and C++/CUDA programming
• Proficiency in profiling live training and inference runs
• Expertise in low-precision numerics or kernel authoring Drive innovation in Physical AI through expertise in PyTorch, distributed systems, and cutting-edge optimizations.