Software Engineer - Machine Learning Compiler
About the Role:
We are seeking a highly skilled Software Engineer to join our team and contribute to the development of machine learning compilers. As a member of our team, you will be responsible for evaluating and improving the performance of various machine learning frameworks.
Key Responsibilities:
* Evaluate and improve the performance of different machine learning frameworks using knowledge of compilers.
* Apply advanced graph optimization techniques to maximize the utilization of GPU hardware resources.
* Write and review code for machine learning graph compilers and runtimes.
Requirements:
To be successful in this role, you will need:
* A bachelor's degree in Computer Science, Computer Engineering, or a related field.
* Strong C/C++ programming skills and experience with LLVM, SYCL, CUDA, or OpenCL.
* Familiarity with data structures and algorithms, including object-oriented programming.
* At least 2 years of industry experience in software engineering.
Preferred Skills:
* Knowledge of GPU architecture and optimization techniques for GPGPU code.
* Experience with a Machine Learning Graph Compiler.
* Familiarity with MLIR.
Benefits:
* A competitive salary and performance-related bonus.
* Maternity/paternity leave and education assistance.
* Relocation and immigration support.
* A comprehensive benefits package including life, medical, income, and travel insurance.
Culture:
We value diversity, equality, and inclusion in the workplace. We believe that every individual brings unique perspectives and experiences that contribute to our success. If you share our values and are passionate about technology, we encourage you to apply for this exciting opportunity.