Machine Learning Engineer

Roche Holding AG; Basel; Vor Ort; CHF 120.000 - 180.000; Vollzeit

Zusammenfassung

Roche Holding AG in Basel sucht eine/n Machine Learning Engineer, um Backend-Komponenten und Tools für wissenschaftliche Agenten zu entwickeln. Sie schreiben sauberen Code, integrieren Modelle in eine API und unterstützen skalierbare Arbeitsabläufe.

Sie arbeiten eng mit Data Scientists zusammen, bauen Python-Wrappers, REST-APIs mit FastAPI/Flask, entwickeln Data-Pipelines für Retrieval-Augmented Generation und pflegen Docker/CIs.

Qualifikationen

• Abgeschlossenes Hochschulstudium (BS/MS oder PhD) mit starkem Software-Engineering-Background.
• Fundierte Python-Kenntnisse, Erfahrung mit Git, Testing, CI/CD sowie sauberer Code.
• Erfahrung in der Entwicklung von Anwendungen mit LLMs, Embeddings, Prompting und Funktionsaufrufen.

Aufgaben

• Backend-Komponenten und Tools für wissenschaftliche Agenten entwickeln.
• Python-Wrappers und API-Definitionen erstellen, damit Modelle Aufgaben ausführen können.
• Backend-Services mit FastAPI/Flask implementieren, um Agenten-Funktionen bereitzustellen.

Kenntnisse

Python

Git

CI/CD

Testing

LLM-Apps

Ausbildung

BS/MS

PhD

Tools

FastAPI

Flask

Docker

Kubernetes

Jobbeschreibung

Bei Roche kannst du ganz du selbst sein und wirst für deine einzigartigen Qualitäten geschätzt. Unsere Kultur fördert persönlichen Ausdruck, offenen Dialog und echte Verbindungen. Hier wirst du für das, was du bist, wertgeschätzt, akzeptiert und respektiert. Dies schafft ein Umfeld, in dem du sowohl persönlich als auch beruflich wachsen kannst. Gemeinsam wollen wir Krankheiten vorbeugen, stoppen und heilen und sicherstellen, dass jeder Zugang zur Gesundheitsversorgung hat - heute und in Zukunft. Werde Teil von Roche, wo jede Stimme zählt.

Die Position

Our AI for Drug Discovery organization, as part of Roche's Computational Sciences Center of Excellence, advances drug discovery through cutting-edge machine learning. Our Foundation Models team builds large language models (LLMs) and agent platforms that enable next-generation scientific and biomedical applications across the drug-discovery pipeline. We are seeking exceptional scientists to join our efforts on one of the largest scientific agent and automation platforms in the industry. Our team focuses on developing and operating the foundational platform that transforms scientific knowledge and actions from thousands of world-class scientists into sharable, reusable tools, workflows, and agents, reshaping how drug discovery operates with large-scale in-house scientific use cases.

We are seeking a Machine Learning Engineer to develop the backend components and tooling for our scientific agents. In this role, you will build the software layer that connects our models to internal scientific capabilities. You will focus on writing clean, performant code that wraps scientific logic into accessible APIs and maintaining the infrastructure that supports our agent workflows.

The Opportunity

• Develop scientific tools by creating Python wrappers and API definitions that allow models to execute specific tasks (e.g., molecular property calculation, database queries).

• Build backend services using modern web frameworks (FastAPI/Flask) to serve agent capabilities to internal platforms.

• Implement data pipelines for Retrieval Augmented Generation (RAG), ensuring models have access to up-to-date proprietary scientific data.

• Maintain infrastructure by writing Dockerfiles, setting up CI/CD pipelines, and ensuring automated testing for agent components.

• Integrate models into the execution environment, handling inference requests and validating structured outputs.

• Write high-quality code that is modular, well-tested, and adheres to the team’s software engineering standards.

• Collaborate closely with computational scientists and subject matter experts on designing and evaluating targeted agents for drug discovery.

• Contribute to frontier research related to agentic use in scientific scenarios.

Who You Are

• Education: BS/MS with 1-4+ years of experience, or PhD graduates with strong software engineering skills.

• Software Foundation: Strong proficiency in Python and standard software engineering practices (Git, testing, CI/CD).

• LLM Application Development: Experience building applications that utilize LLMs, with an understanding of embeddings, prompt engineering, and function calling.

• Modern Stack: Familiarity with containerization and backend development frameworks.

• Execution Focused: Ability to work independently on defined engineering tasks and deliver robust solutions.

• Collaborative: Eager to learn from senior engineers and scientists and contribute to a fast-paced technical environment.

• Mission Driven: A strong interest in applying software engineering to drug discovery and a commitment to patient impact.

• Experience with molecular modalities (e.g., protein sequences, chemical graphs) or scientific data is a plus, but not required.

• General interest in the natural sciences (chemistry, biology, or physics).

Wer wir sind

Eine gesündere Zukunft treibt uns zur Innovation an. Mehr als 100.000 Mitarbeiter weltweit arbeiten gemeinsam daran, wissenschaftliche Fortschritte zu erzielen und sicherzustellen, dass jeder Zugang zur Gesundheitsversorgung hat - heute und für zukünftige Generationen. Durch unser Engagement werden über 26 Millionen Menschen mit unseren Medikamenten behandelt und mehr als 30 Milliarden Tests mit unseren Diagnostik-Produkten durchgeführt. Wir ermutigen uns gegenseitig, neue Möglichkeiten zu erkunden, Kreativität zu fördern und hohe Ziele zu setzen, um lebensverändernde Gesundheitslösungen zu liefern.

Gemeinsam können wir eine gesündere Zukunft gestalten.

Roche ist ein Arbeitgeber, der die Chancengleichheit fördert.