Senior Machine Learning Infrastructure Engineer (Precision, Diagnostics & Hardware)
Veeda AI
Job Description
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 Systems & Scalability: Design, optimize, and maintain high-throughput distributed training systems across large-scale GPU clusters for multi-modal foundation models.
Precision & Numerical Stability: Debug, diagnose, and resolve subtle numerical instability issues (underflow/overflow, loss spikes, gradient explosion, and mixed-precision divergence) in FP16, BF16, FP8, and custom quantization schemes.