Staff Machine Learning Engineer (f/m/d)

Orbem uses AI to industrialize MRI, delivering non-invasive insight into everything from fruits to eggs to the human body. We are transforming what is becoming the world's largest biological dataset into actionable intelligence to reduce waste, increase quality, and accelerate the shift toward a more sustainable and healthier future.

Headquartered and founded in Munich, Germany, with an office now in Houston, Texas, our world-class team is scaling inside-out technology to transform how humanity sees and understands biological matter.

Help Us See What's Possible. Join Us.

Your Role

As a Staff Machine Learning Engineer, you will be at the forefront of building the intelligent systems that power Orbem’s next-generation technology. You’ll design and deliver scalable, production ready ML solutions that move seamlessly from prototype to real-world deployment. Partnering closely with data scientists and software engineers across teams, you’ll shape the core architecture that brings breakthrough ideas to life.

You’ll be a driving force for excellence, defining best practices, elevating engineering standards, and championing high-quality machine learning at every stage. With your production mindset, passion for impact, and vision for what’s possible, you’ll help unlock the next era of AI-powered MRI imaging propelling Orbem toward its ambition of becoming a true Machine Learning powerhouse.

Your Day-to-Day

On a typical day you would:

• lead design and implementation of ML systems.
• build and improve code for training ML models, prioritizing modularity, scalability and robustness.
• collaborate with data scientists in training and fine-tuning ML models.
• optimize and troubleshoot production ML code and models, focusing on latency and memory footprint.
• deploy and monitor ML models in production, ensuring model stability.
• grow the know-how in production ML tools and concepts within the team.
• guide the team in adopting the best ML engineering practices, ensuring a consistent and production-ready approach across all AI developments.

Machine Learning

• You have deep knowledge of pytorch and distributed training.
• You are experienced in ML system design.
• You bring experience in optimizing model architectures for both efficient inference and training.
• You have a history of building and automating complex data pipelines for ML.
• You are familiar with tools for ML in production and model deployment.

• You have strong python coding skills, you are an advocate and expert in best practices and design patterns.
• You have experience with version control tools and tools for creating reproducible, containerized environments.
• You are well versed in CI/CD pipelines.
• You have experience working with Cloud providers to scale and automate ML workflows.

• Proven technical leadership on cross-functional initiatives, successfully guiding complex projects that span multiple engineering teams.
• Strategic influence in shaping the long-term technical roadmap for a product or platform, ensuring engineering decisions align with business goals.
• High technical judgment and a pragmatic approach to decision-making, with a demonstrated ability to balance deep technical work with product deadlines.

• You have experience with CUDA and GPUs to facilitate accelerated computing.
• You have worked with 3D data and are familiar with related computational challenges.
• You worked with ML in healthcare, and are familiar with data privacy and security challenges and techniques like federated learning.

• We Own Every Challenge: we enjoy complexity and thrive under uncertainty.
• We Strive for Better: we seize any opportunity for growth and challenge the status quo. We are constantly learning and improving.
• We Imagine New Frontiers: we think beyond “doable” and “reasonable”. We design a sustainable and healthy future together.

International Environment:

• Join a team with 40+ nationalities across 5 continents, all driven by a shared purpose: shedding light on the world’s toughest challenges.

• Stock Options: Share in Orbem’s success.
• Relocation Support: Seamless support for your move to Germany.
• Learning & Development: €1,750 annual budget for personal growth.
• Fitness Membership: Access to Urban Sports Club or Wellpass.
• Childcare Reimbursement: Support for Kita/Kindergarten fees.
• Deutschland Ticket: Full coverage of public transportation.

• Flexible Hours & Home Office: Work when and where it suits you.
• WFH Perks: Get €185 to upgrade your home office + an external monitor on us!
• 30 Days Paid Leave: Plenty of time to recharge.
• Personal Leave: Flexibility for life’s important moments.
• Work From Anywhere: Experience new cultures and environments for up to 60 days per year.

Join an ambitious, fast-growing team working on breakthrough technology. In our scale-up environment, you'll have the freedom to lead your projects and make an impact. We provide a platform for you to explore, innovate, and define your vision for the future. At Orbem, we're committed to helping you discover your strengths, and while we aim to teach you, we also want to learn from you.

Your Team:

You'll become part of our diverse and international team. Learn more about the team members, their work, and challenges here: https://orbem.ai/company/

At Orbem, we're committed to building a smart, diverse team, and we recognize that self-doubt can prevent talented individuals from applying. If you feel you don't meet every requirement, we'd love to hear from you anyway!

Compensation Range: €90K - €100K