Machine Learning Compiler Engineer
Job Title: Machine Learning Compiler Engineer
About the Role
Do you have a passion for both compiler engineering and machine learning? As a Machine Learning Compiler Engineer, you’ll be in a unique position to design and develop the next generation of compilers that power AI-driven applications. You’ll help optimize deep learning models and algorithms, ensuring they run efficiently across a variety of hardware platforms—whether it’s GPUs, specialized accelerators, or custom hardware.
In this role, you’ll work at the crossroads of AI and high-performance computing, transforming cutting-edge machine learning models into optimized code that accelerates training, inference, and deployment. You’ll be part of a team that pushes the boundaries of what’s possible, working alongside brilliant engineers and researchers to drive the performance and scalability of our AI-powered solutions.
What You’ll Do
Design and Implement ML Compiler Optimizations: Work on developing compilers specifically tailored for machine learning, focusing on optimizing model performance and making machine learning workflows faster and more efficient.
Optimize Across Hardware Platforms: Focus on improving execution across different hardware platforms, from GPUs to TPUs, CPUs, and emerging accelerators, ensuring that machine learning models run optimally on a variety of devices.
Integrate with ML Frameworks: Work closely with popular machine learning frameworks (TensorFlow, PyTorch, etc.) to integrate and optimize their models for faster computation and lower energy consumption.
Enhance Compiler Toolchains: Continuously improve existing compiler toolchains, adding features and enhancements that support advanced machine learning operations, ensuring seamless integration with cutting-edge hardware.
Collaborate with Cross-Functional Teams: Collaborate with researchers, hardware engineers, and software developers to ensure that compilers meet the specific needs of AI models and optimize for performance and scalability.
Work on Real-World AI Applications: Apply your expertise to real-world machine learning applications, solving challenges related to scaling AI workloads, improving latency, and enhancing the computational efficiency of deep learning models.
Drive Innovation: Stay on the cutting edge of AI and compiler technology, experimenting with new techniques to push the performance of machine learning systems.
Who You Are
You are an innovative problem solver who has a deep understanding of both machine learning and compiler technology. You bring:
Extensive Compiler Expertise:
5+ years of experience in compiler development, optimization, or related fields, with a focus on high-performance systems.
Familiarity with LLVM, Clang, and other compiler frameworks, and experience implementing optimizations for machine learning tasks.
Machine Learning Experience:
Strong understanding of machine learning models and algorithms, including deep learning, reinforcement learning, and other AI techniques.
Practical experience working with machine learning frameworks (TensorFlow, PyTorch, ONNX, etc.) and deploying models across different hardware platforms.
Strong Programming Skills:
Proficiency in C++, Python, and other relevant programming languages, with experience in writing efficient, production-quality code.
Experience working with ML libraries like TensorFlow, PyTorch, scikit-learn, or Keras, and optimizing them for various hardware accelerators.
High-Performance Computing Knowledge:
Experience in performance optimization and tuning for complex systems, including parallel computing and GPU/TPU optimizations.
Familiarity with multi-threading, SIMD, and hardware acceleration techniques used in AI workloads.
Collaborative and Effective Communicator:
Excellent communication skills to work with cross-functional teams, from researchers to software developers, and to explain complex technical concepts to non-experts.
A team player who thrives in collaborative environments, sharing ideas and solutions to drive success.
Why Join Us?
Work at the Forefront of AI and Performance Optimization: Contribute to groundbreaking work in optimizing AI workloads and improving the efficiency of machine learning models across diverse hardware platforms.
Grow with a Talented Team: Work with some of the brightest minds in AI, compiler engineering, and hardware design, with ample opportunities for mentorship, learning, and career advancement.
Competitive Compensation and Benefits: Receive a competitive salary, stock options, and a comprehensive benefits package, including healthcare, wellness programs, and retirement plans.
Innovative and Supportive Culture: Join an organization that fosters innovation, creativity, and collaboration, where your ideas and contributions will shape the future of AI optimization.
Work-Life Balance: Enjoy flexible hours, remote work options, and a work culture that values balance and flexibility, ensuring you thrive both personally and professionally.
Key Skills: