AI Frameworks Integration Lead
Description
Invent the future with us.
Recognized by Fast Company’s 2023 100 Best Workplaces for Innovators List, Ampere is a semiconductor design company for a new era, leading the future of computing with an innovative approach to CPU design focused on high-performance, energy efficient, sustainable cloud computing.
By providing a new level of predictable performance, efficiency, and sustainability Ampere is working with leading cloud suppliers and a growing partner ecosystem to deliver cloud instances, servers and embedded/edge products that can handle the compute demands of today and tomorrow.
Join us at Ampere and work alongside a passionate and growing team — we’d love to have you apply!
About the role:
As the AI Frameworks Integration Engineering Lead at Ampere, you'll build and lead a team integrating cutting-edge AI frameworks, working closely with our silicon design experts to develop and enhance machine learning libraries. You'll also analyze neural network models, proposing critical improvements and optimizations to drive the performance of next generation AI compute technologies.
What you’ll achieve:
- Lead the AI Framework Software Integration Team: Head a dynamic team dedicated to integrating advanced AI frameworks, driving cutting-edge development and innovation.
- Collaborate with the Silicon Design Team: Work closely with Ampere's silicon design experts on the groundbreaking Ampere Aurora AI accelerator, ensuring seamless hardware-software synergy.
- Implement and Optimize Machine Learning Libraries: Design and implement innovative solutions and enhancements for popular machine learning libraries such as Llama.cpp, PyTorch, JAX, vLLM, and ONNX, significantly improving their functionality and efficiency.
- Analyze and Optimize Neural Network Models: Conduct comprehensive analyses of neural network models, identifying and proposing critical improvements and optimizations at both the model and framework levels, advancing performance and effectiveness.
- Develop Cutting-Edge AI Co-Processors/Accelerators: Take the lead in building state-of-the-art software and hardware AI co-processors/accelerators, pushing the boundaries of what's possible in AI technology.
About you:
- Bachelor's degree in Computer Science, Electrical Engineering, Mathematics, or a similar quantitative field (Masters/PhD is preferred) and a minimum 12 years of overall software engineering experience including 4 years of experience directly related to this role.
- Proficiency in Python and C/C++: With at least 4 years of hands-on experience, you excel in these critical programming languages, supporting robust and efficient software development.
- Expertise in Large-Scale Software Systems: Your previous experience with complex software architectures showcases your ability to handle big, challenging projects.
- Mastery of Essential Tools: You are proficient with Linux, Git, GCC, and LLVM, ensuring seamless integration and development across various platforms.
- Startup Agility: Thriving in fast-paced startup environments, you are adept at turning ideas into reality and achieving results swiftly.
- Machine Learning and Deep Learning Acumen: You bring strong knowledge of key concepts and have practical experience with popular frameworks like TensorFlow, PyTorch, Llama.cpp, and vLLM, positioning you at the forefront of AI innovation.
What we’ll offer:
At Ampere we believe in taking care of our employees and providing a competitive total rewards package that includes base pay, bonus, equity, and comprehensive benefits. We offer an annual bonus program tied to internal company goals and annual meritocratic equity awards that enable our employees to participate in the success of the company.
Benefits highlights include:
- Premium medical health care, so that you and your family members can feel secure in your health
- A generous paid time off policy so that you can embrace a healthy work-life balance
- A wide variety of office amenities including nutritious snacks and refreshing drinks, free gym and sauna access to keep you fueled and healthy
- Flexible working hours and a remote work policy that includes reimbursement of connectivity costs and equipment to work from home
- Sports card fully financed by Ampere
And there is much more than compensation and benefits. At Ampere, we foster an inclusive culture that empowers our employees to do more and grow more. We are passionate about inventing industry leading cloud-native designs that contribute to a more sustainable future. We are excited to share more about our career opportunities with you through the interview process.
#LI-DR
#LI-Remote