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