Senior/Staff ML Engineer, AI-Assisted Efficiency Optimization
- Implement AI-driven methods for GPU kernel optimizations to enhance program efficiency and performance leveraging tools such as CompilerGym, KernelLLM, etc.
- Develop strategies for resource-efficient deployment of intelligent optimization processes using frameworks such as AutoFDO.
- Guide high-level optimizations that include both highly optimized kernels, as well as AI-driven neural architecture search (NAS) to optimize training efficiency and inference latency.
- Assess performance improvements via evaluation metrics and real-world feedback.
- Collaborate with internal teams to benchmark processes and strategies for future AI-assisted optimizations.
- Utilize state-of-the-art leaderboard scoring systems to test the ability of different AI models to generate efficient GPU kernels.
- Stay current with industry advances.
- Bachelor's or Master's Degree in Computer Science, Engineering, or a related field.
- Extensive experience with AI models, including but not limited to LLMs.
- Strong understanding of retrieval-augmented generation (RAG) and Language Models (LLMs).
- Ability to self-motivate, undertake complex assignments independently, and while enjoying to balance innovative exploration with practical considerations.
- Experience with GPU programming, CUDA, Triton, and Machine Learning algorithms.
- Familiarity with optimization techniques such as neural architecture search (NAS).
- Hands-on experience with optimization programs, such as Google's AutoFDO.
Recommended Jobs
Staff Machine Learning Engineer (CMG Engineering)
We exist to wow our customers. We know we’re doing the right thing when we hear our customers say, "How did we ever live without Coupang?" Born out of an obsession to make shopping, eating, and livin…
Receptionist $20/hr + Benefits State Farm Insurance Agency
Full-Time Receptionist Location: Morgan Hill, CA Pay: $20/hour + Bonuses & Commissions About the Role: Were looking for a friendly and reliable Receptionist to join the Gabriela Ruvalcaba S…
Staff Backend Engineer
Who We Are Tonal is the world’s first all-in-one home gym with a simply stunning design. It has completely revolutionized the fitness journey, with adaptive weight and coaching cues powered by adv…
Investment Advisor
Flexibility and Control. And a business partner who gives you the freedom to have both. You’ve spent a good portion of your career building the practice you envisioned. And for whatever reason,…
Director of Customer Success
Title: Director of Customer Success (Find Patients) Team: Customer Success Location: San Diego, CA - Hybrid - In office 3x per week Reports To: Senior Director, Customer Success Abou…
Service Readiness Project Manager
What To Expect The Tesla Service team is seeking a highly motivated Service Readiness Project Manager to support our mission of accelerating the world's transition to sustainable energy. In this r…
Law Firm Full Service Bookkeeper
"IMPORTANT: TO APPLY" To ensure we consider only the most dedicated and detail-oriented professionals, please follow these instructions when applying: • Call us at 213-289-0619 to express your interes…
Product Manager, Search & Discovery
Step into the innovative world of LG Electronics. As a global leader in technology, LG Electronics is dedicated to creating innovative solutions for a better life. Our brand promise, 'Life's Good', e…
💻 Sales Representative - Flexible Hours - Work from Home
Location: Remote Position/Work from Home Job Type: Full-Time/Part-Time Compensation: 100% Commission About Us: GIA Legacy Planning is a dynamic, client-focused insurance agency that s…
Assistant Store Manager PT
About Our Company Lids Sports Group is the largest licensed sports retailer in North America, selling fan and fashion-oriented headwear and apparel across the US, Canada, Mexico, Europe, and Austr…