Machine Learning Infrastructure Engineer - Remote

Overview

The Diffuse Project is seeking a Machine Learning Infrastructure Engineer to lead the development of robust, scalable backend systems that power machine learning–driven discoveries in structural biology.

In Short

• Architect, build, and maintain ML infrastructure pipelines.
• Design and manage data ingestion and preprocessing workflows.
• Develop and maintain backend services and APIs.
• Support GPU/accelerated training on HPC clusters or cloud platforms.
• Implement data versioning, model tracking, and reproducibility tools.
• Collaborate with ML researchers and experimentalists.
• Work effectively in a multidisciplinary team environment.

Requirements

• Strong programming skills in Python, ideally with PyTorch experience.
• Deep understanding of machine learning infrastructure.
• Proficiency in backend development and containerization.
• Experience with distributed compute environments.
• Solid understanding of scientific computing workflows.
• At least two years of experience working on ML models.
• (Bonus) Familiarity with structural biology data formats.
• (Bonus) Experience designing systems for diffusion-based models.

Benefits

• Competitive compensation package commensurate with experience.
• Potential for extension based on performance and business needs.
• Collaborative and product-minded work environment.