At a Glance
- Tasks: Contribute to groundbreaking Physical AI innovations and optimize distributed training.
- Company: Join Veeda AI, a leader in multimodal foundation models.
- Benefits: Competitive salary, flexible work options, and opportunities for professional growth.
- Other info: Dynamic team environment with exciting challenges and career advancement.
- Why this job: Be at the forefront of AI technology and make a real impact.
- Qualifications: Bachelor's degree and hands-on experience with PyTorch required.
The predicted salary is between 135000 - 165000 $ per year.
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 Py Torch 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 Py Torch 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 Py Torch, distributed systems, and cutting-edge optimizations. #J-18808-Ljbffr
Technical Staff Member for Physical AI in Toronto employer: Veeda AI
Veeda AI is an exceptional employer located in the vibrant city of Toronto, offering a dynamic work culture that fosters innovation and collaboration. Employees benefit from opportunities for professional growth through hands-on projects in cutting-edge robotics, while enjoying a supportive environment that values creativity and teamwork. With a focus on advancing Physical AI, Veeda AI provides a unique chance to make a meaningful impact in the field of robotics.
