Staff Software Engineer, ML Acceleration

📋 Description

• Analyze ML models to identify and fix performance bottlenecks.
• Incorporate OSS tools to empower ML engineers to profile and optimize models.
• Deliver solutions to streamline model deployment across hardware platforms.
• Collaborate with ML researchers to balance accuracy and speed.
• Implement optimizations with CUDA, Triton, and custom kernels.
• Promote engineering excellence within the team.

🎯 Requirements

• Education: Bachelor's or Master's in CS/Engineering or related field.
• Experience: 5+ years, GPU programming and optimization.
• Skills: C++, Python, CUDA, Triton, PyTorch, TensorRT, ONNX, GPU kernels.
• Problem-Solving: Strong analytical skills.
• Communication: Excellent verbal and written skills.
• AV experience: Autonomous vehicles experience bonus.

🎁 Benefits

• Equal opportunity workplace
• Inclusive culture and innovation