Backend ML Engineer for Scientific AI in Drug Discovery

Roche; Basel; Vor Ort; CHF 110.000 - 160.000; Vollzeit

Zusammenfassung

Roche in Basel is seeking a Machine Learning Engineer to develop backend components and tooling for scientific agents. You will write clean, performant Python that wraps complex logic into accessible APIs and maintain infrastructure supporting agent workflows.

You will build Python wrappers, API definitions, and data pipelines for Retrieval Augmented Generation, collaborate with computational scientists, and contribute to cutting‑edge agent platforms that power drug discovery.

Qualifikationen

• BS/MS with 1–4+ years of experience, or PhD graduates with strong software engineering skills.
• Strong proficiency in Python and standard software engineering practices (Git, testing, CI/CD).
• Experience building applications that utilize LLMs, with embeddings, prompt engineering, and function calling.
• Familiarity with containerization and backend development frameworks.
• Ability to work independently on defined engineering tasks and deliver robust solutions.

Aufgaben

• 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.

Kenntnisse

Python

LLM development

API design

CI/CD

Testing

Ausbildung

BS/MS or PhD in relevant field

Tools

Docker

FastAPI

Flask

Git

Jobbeschreibung

Roche in Basel is seeking a Machine Learning Engineer to develop backend components and tooling for scientific agents. You will write clean, performant Python that wraps complex logic into accessible APIs and maintain infrastructure supporting agent workflows.

You will build Python wrappers, API definitions, and data pipelines for Retrieval Augmented Generation, collaborate with computational scientists, and contribute to cutting‑edge agent platforms that power drug discovery.