Machine Learning Engineer

About The Company

Orbem GmbH is an impact-driven deep-tech scaleup with a global presence, founded in Munich, Germany, and now expanding internationally with its newest office in Houston, Texas. The company specializes in developing fast, accurate, and accessible imaging solutions that unlock sources of knowledge previously unattainable. Orbem's mission is to make a meaningful difference by creating innovative solutions to sustainably feed the world, accelerate the transition to a green economy, and transform disease detection. Through cutting-edge AI-powered imaging technology, Orbem aims to revolutionize industries and improve lives worldwide.

About The Role

We are seeking a highly skilled and motivated Staff Machine Learning Engineer to join our dynamic team. In this pivotal role, you will be at the forefront of building intelligent systems that drive Orbem's next-generation imaging technology. Your primary responsibility will be designing and delivering scalable, production-ready machine learning solutions that seamlessly transition from prototypes to real-world applications. Collaborating closely with data scientists and software engineers across various teams, you will shape the core architecture that transforms breakthrough ideas into tangible products.

The ideal candidate will be a driving force for engineering excellence, setting best practices, elevating standards, and championing high-quality machine learning initiatives at every stage. With a production mindset, passion for impact, and a visionary approach, you will help unlock the next era of AI-powered MRI imaging, propelling Orbem towards becoming a leading machine learning powerhouse in the industry.

Qualifications

The successful candidate will possess a strong technical background and align with Orbem's core values of ownership, continuous improvement, and innovation. Key qualifications include:

• Deep expertise in PyTorch and distributed training frameworks.
• Proven experience in ML system design, including optimizing model architectures for efficient inference and training.
• Experience building and automating complex data pipelines for machine learning workflows.
• Familiarity with tools and platforms for ML deployment and production environments.
• Strong Python programming skills, with a focus on best practices and design patterns.
• Experience with version control systems and creating reproducible, containerized environments.
• Knowledge of CI/CD pipelines and cloud-based scaling and automation of ML workflows.
• Demonstrated leadership in guiding cross-functional projects and influencing technical roadmaps.
• Experience working with CUDA, GPUs, and 3D data, especially in healthcare applications, is highly desirable.
• Understanding of data privacy, security, and techniques like federated learning.

Responsibilities

As a Staff Machine Learning Engineer, your day-to-day responsibilities will include:

• Leading the design and implementation of scalable ML systems that meet production standards.
• Developing and refining codebases for training ML models, emphasizing modularity, robustness, and scalability.
• Collaborating with data scientists to train, fine-tune, and optimize machine learning models.
• Optimizing and troubleshooting ML models in production, focusing on reducing latency and memory footprint.
• Deploying and continuously monitoring ML models to ensure stability and performance in real-world environments.
• Growing the team's expertise in production ML tools, frameworks, and best practices.
• Guiding and mentoring team members to adopt high-quality ML engineering standards, fostering a culture of excellence and innovation.

Benefits

Orbem offers a comprehensive package of benefits designed to support your professional growth and personal well-being, including:

• International environment with over 40 nationalities across five continents, driven by a shared purpose of addressing global challenges.
• Attractive compensation package, including stock options that allow you to share in Orbem's success.
• Visa and relocation support to facilitate your move to Germany.
• Annual learning and development budget of €1,750 to pursue personal and professional growth.
• Access to fitness memberships such as Urban Sports Club or Wellpass.
• Childcare reimbursement to support families with kindergarten or Kita fees.
• Full coverage of public transportation costs through the Deutschland Ticket.
• Flexible work hours and hybrid work model, allowing you to work when and where it suits you best.
• Home office upgrade allowance of €185 and an external monitor to enhance your remote work setup.
• 30 days of paid leave annually, with additional personal leave options.
• Opportunity to work from anywhere for up to 60 days per year, enabling cultural exploration and diverse work environments.

Joining Orbem means becoming part of a fast-growing, innovative team where your contributions can make a significant impact. We foster an environment of exploration, creativity, and continuous learning, providing a platform for you to lead projects and shape the future of AI-powered imaging technology.

Equal Opportunity

Orbem GmbH is committed to creating an inclusive and diverse workplace. We welcome applications from individuals of all backgrounds, regardless of race, gender, age, religion, or disability. We believe that diverse teams foster innovation and excellence, and we are dedicated to providing equal opportunities for all employees. If you are passionate about making a difference and meet the qualifications outlined, we encourage you to apply, even if you feel you do not meet every requirement. Your unique perspective and experiences are valuable to us.