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 trainingProfile latency and GPU utilization for improved throughputEnhance precision and stability for video tokenizer and VAE layersWrite and tune custom CUDA and Triton kernelsDevelop detection layer for silent data corruption issuesRequirements:Bachelor's degree or equivalent in a technical fieldHands-on experience with PyTorch in multi-node jobsFluency in Python and C++/CUDA programmingProficiency in profiling live training and inference runsExpertise in low-precision numerics or kernel authoringDrive innovation in Physical AI through expertise in PyTorch, distributed systems, and cutting-edge optimizations.#J-18808-Ljbffr

📌 Technical Staff Member For Generative Modeling (Toronto)
🏢 Veeda AI
📍 Toronto