ML Platform Engineer for AI-Driven Drug Discovery

Proxima; Town of Montana (WI); Presencial; USD 120.000 - 160.000; Jornada completa

Descripción de la vacante

Proxima is seeking a versatile Machine Learning Engineer to build and optimize software for drug discovery. You will work across automation, infrastructure, data, and ML with Python and PyTorch at the core, in a Kubernetes-driven environment.

You will collaborate with researchers and infrastructure engineers to deploy models, run experiments, and automate workflows. A strong ownership mindset and ability to learn across the stack are valued.

Formación

• Strong Python and ML workflow experience.
• Experience with ML or scientific computing pipelines.
• Familiarity with containerized workloads and CI/CD.

Responsabilidades

• Build reliable scientific workflows that colleagues can run, reproduce, and troubleshoot with less manual intervention.
• Integrate new models, datasets, and evaluation methods while preserving data correctness and compatibility with existing experiments.
• Improve compute workflows through better job submission, monitoring, failure recovery, and artifact tracking.
• Identify and remove bottlenecks in data preparation, execution, and evaluation so researchers can iterate faster.
• Deliver libraries, services, and developer tools that are easy to use and maintain, with tests, docs, and monitoring.
• Contribute to the culture of a rapidly growing company.
• Be challenged by colleagues and learn something new every day.

Conocimientos

Python

PyTorch

Linux

CI/CD

Kubernetes

Cloud platforms

Data pipelines

Herramientas

Docker

CI tooling

Descripción del empleo

Proxima is seeking a versatile Machine Learning Engineer to build and optimize software for drug discovery. You will work across automation, infrastructure, data, and ML with Python and PyTorch at the core, in a Kubernetes-driven environment.

You will collaborate with researchers and infrastructure engineers to deploy models, run experiments, and automate workflows. A strong ownership mindset and ability to learn across the stack are valued.