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.
Track your applications on mobile The free Whileresume app, on iPhone and Android.
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 inYou might also like these jobs
No closely matching jobs yet — here are the most recent ones.