The role of Machine Learning Algorithm/SW Optimization Engineer involves the development and optimization of advanced AI compute solutions. As part of the AI Compute department, you will work on innovative AI models, focusing on machine learning algorithm optimization techniques that bridge the gap between software and hardware.
Responsibilities
* You will be responsible for understanding and optimizing performance of deep learning libraries through in-depth knowledge of AI frameworks, algorithms, models, and related hardware.
* Setting up AI workloads, analyzing and developing optimization techniques for existing and future compute architecture in collaboration with different stakeholders is a key aspect of this role.
* Explore new algorithms/optimizations for custom architectures to enable SW-HW co-design.
* Staying up-to-date with the latest deep learning literature to implement state-of-the-art algorithms is essential.
* Following up on progress within research projects and communicating the accomplished/in-process work to internal and external stakeholders and adapting where needed.
In the first few years, you will be an integral part of our ongoing EuroHPC DARE project that aims to drive Europe's independence in high-performance computing and AI.
Requirements
* A PhD in Computer Science, Engineering, Mathematics, or equivalent is required.
* Possessing a strong background with parallel programming and proven ability of CUDA and Python programming is necessary.
* Experience with modern Deep Learning Frameworks (PyTorch, Tensorflow, Jax) is a must.
* Experience with distributed training frameworks (e.g., Ray, Dask, PyTorch Lightning) is also required.
* Strong background in algorithmic optimization techniques like quantization, pruning, and distributed training methodologies.
* Experience in software performance analysis and optimization.
* Prior experience with optimizing AI for AI Compute Architectures is a plus.
* Understanding of performance analysis tools and hardware optimization techniques.
* Fluent in English, both speaking and writing.
Benefits
This position offers the opportunity to join one of the world's premier research centers in nanotechnology at its headquarters in Leuven, Belgium.
We are committed to being an inclusive employer and proud of our open, multicultural, and informal working environment with ample possibilities to take initiative and show responsibility.
We commit to supporting and guiding you in this process; not only with words but also with tangible actions.
Through imec.academy, 'our corporate university', we actively invest in your development to further your technical and personal growth.