Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization

As a Senior/Staff ML Engineer, you will lead GenAI-assisted GPU kernel optimization delivery, targeting peak FLOPs and reduced latency. Implement AI-driven methods for GPU kernel optimization using tools like CompilerGym and KernelLLM. Develop resource-efficient deployment strategies for intelligent optimization using AutoFDO. Guide high-level optimizations including optimized kernels and AI-driven NAS to improve training efficiency and inference latency. Evaluate performance with metrics and real-world feedback; collaborate with internal teams to benchmark AI-assisted optimizations. Stay current with industry advances and participate in leaderboard scoring to test AI models' ability to generate efficient GPU kernels.