Machine Learning Engineer (LUMI AI Factory)

Would you like to develop software for the highest level of AI supercomputing?

LUMI AI Factory powers the most ambitious development projects ranging from sensor fusion of physical AI system to large public collaborations to develop the next generation of European large language models.

At CSC, we offer unparalleled growth opportunities, a supportive and inclusive environment, and the chance to work on the future of research and learning. Join us today!

We are now looking for a Machine Learning Engineer to develop and maintain containerised AI software environments, providing LUMI AI Factory customers access to all the tools they need for running demanding AI workloads. You track user needs and the development of core AI libraries in order to gain a strong understanding of the AI software ecosystem. Based on this understanding, you work to deliver a comprehensive environment optimized for the LUMI supercomputer and, in the future, LUMI-AI supercomputer.

You will be part of a small technical team focused on developing and providing a relevant tool offering. The current software environment is comprised of a collection of PyTorch-based AI containers for the LUMI supercomputer. As a Machine Learning Engineer, you extend the coverage of the environment to additional deep learning frameworks, such as TensorFlow and JAX, as well as mixed workflows involving both deep learning as well as computational simulation libraries. The work will also include creating and implementing maintenance practices for these tools. A major part of the work is debugging issues resulting from a mix of quickly evolving AI libraries and a unique supercomputing environment. Part of your duties will also be interacting with the user support team as well as the technical teams in other international AI Factory partner organizations.

You also participate in designing containerised workflows to streamline the use of AI on supercomputing systems. Containerised workflows combine pre-built containe