Machine Learning Algorithm/SW Optimization Engineer
About the Role:
We are seeking a highly skilled Machine Learning Algorithm/SW Optimization Engineer to join our team. As a key member of our research group, you will be responsible for optimizing performance of deep learning libraries through deep knowledge of AI frameworks, algorithms, models, and related hardware.
Key Responsibilities:
* Understand and optimize performance of deep learning libraries through deep knowledge of AI frameworks, algorithms, models, and related hardware.
* Set up AI workloads, analyze, and develop optimization techniques for existing and future compute architectures in collaboration with stakeholders.
* Explore new algorithms/optimizations for custom architectures to enable SW-HW co-design.
* Stay current with deep learning literature to implement state-of-the-art algorithms.
* Communicate progress within research projects to stakeholders and adapt as needed.
* Participate in the EuroHPC DARE project to advance Europe's independence in high-performance computing and AI.
Requirements:
* PhD in Computer Science, Engineering, Mathematics, or related field.
* Proficient in parallel programming, CUDA, Python.
* Experience with Deep Learning Frameworks (PyTorch, TensorFlow, Jax).
* Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning).
* Knowledge of optimization techniques like quantization, pruning, distributed training.
* Experience in software performance analysis and hardware optimization.
* Experience with AI compute architecture optimization is a plus.
* Strong communication skills and team-oriented mindset.
* Fluent in English, both spoken and written.
Benefits:
You will join a leading research center in nanotechnology at imec's headquarters in Leuven, Belgium. We foster an inclusive, multicultural, and innovative environment with opportunities for personal and professional growth through imec.academy. We offer a competitive salary with fringe benefits, recognizing your valuable contributions.
About Us:
We are a research-driven organization dedicated to advancing nanotechnology and its applications. Our team is passionate about pushing the boundaries of what is possible and making a meaningful impact on society.