Machine Learning Research Engineer / Machine Learning Research Engineeress

Join the ETH AI Center as a Machine Learning Engineer

We are seeking a skilled Machine Learning Engineer to join our dynamic team. The ideal candidate will be involved in the development, optimization, and maintenance of our codebase used for training large generative neural network models. This role requires a strong background in machine learning, software development, and the ability to work collaboratively in a research-focused environment.

About the Swiss AI Initiative

The Swiss AI Initiative is a collaborative research project led by ETH Zurich and EPFL, focused on developing responsible and transparent generative AI. A significant project of this initiative is the Large Language Model (LLM) effort, which aims to create state-of-the-art language models at various scales, including an ambitious 70B parameter model.

This work leverages the Alps supercomputer at the Swiss National Supercomputing Centre (CSCS), which features over 10,000 NVIDIA Grace Hopper GPUs, making it one of the most powerful AI-focused computing resources in Europe. The Swiss AI Initiative plans to distribute 15-20 million GPU hours annually to support various research and development projects in AI.

Your Role

As a machine learning engineer in this project, you will contribute to the development and optimization of the training pipelines for these large-scale models. This role is at the intersection of cutting-edge research and high-performance computing, advancing Switzerland’s position in AI innovation.

Key responsibilities include:

• Developing and maintaining software for training large-scale neural networks, such as large language models.
• Collaborating with researchers and other engineers to design and implement scalable solutions for model training, evaluation, and deployment.
• Optimizing existing machine learning frameworks to improve performance and efficiency.
• Staying updated with the latest advancements in AI and machine learning technologies.
• Participating in code reviews and maintaining comprehensive documentation to ensure code quality and reproducibility.
• Contributing to research papers and technical reports as needed.

Qualifications

The ideal candidate will possess the following qualifications:

• Master’s degree or higher in Computer Science, Artificial Intelligence, or a related field.
• Proven experience in machine learning and neural network architectures.
• Strong programming skills in Python and familiarity with PyTorch.
• Experience with software development practices, including version control systems, debugging, testing, and deployment.
• Excellent problem-solving abilities and strong analytical skills.
• Ability to work effectively in a team as well as independently.

What We Offer

• A stimulating academic environment at one of the world's leading technical universities.
• The opportunity to work with state-of-the-art supercomputing infrastructure and cutting-edge AI research.
• Collaboration with top researchers and engineers from ETH Zurich, EPFL, CSCS, and other Swiss institutions.
• Flexible working arrangements, including options for remote work.
• Professional development opportunities, including conference attendance and specialized training.
• The chance to contribute to open-source projects with global impact.
• Access to the broader Swiss academic ecosystem and industry partnerships.
• Being part of Switzerland's sovereign AI development, working on technology with national significance.

We look forward to receiving your online application using the form below. Only applications matching the job profile will be considered.

Contact Information

For further information about the ETH AI Center and the Swiss AI Initiative, please visit our website. Questions regarding the position can be directed to Dr. Imanol Schlag at Show e-mail (no applications).