Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
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.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inSimilar openings
Other roles that could suit you.