Machine Learning Infrastructure Engineer
Indexed description
We are looking for a Machine Learning Infrastructure Engineer to join a highly advanced drug discovery platform team working at the intersection of machine learning, large-scale data systems, and computational science. The team builds the core infrastructure that enables AI-driven drug design — processing massive biological and chemical datasets and running large-scale model training on high-performance computing systems.
The mission is to transform cutting-edge research models into reliable, scalable, production-grade systems that directly support the discovery of new medicines. This is a highly impactful environment where engineering excellence meets real-world scientific innovation in the life sciences domain.
We offer a choice of employment form: B2B or Employment Contract
- UoP: 23 000 – 25 000 PLN gross/month
- B2B: 170 – 190 PLN net/hour + VAT
- Build and scale infrastructure for deploying ML models (e.g. GNNs, Transformers) in chemistry and structural biology domains;
- Productionize research-grade models into robust, scalable, and maintainable systems;
- Design and maintain cloud-native data architectures (BigQuery / Snowflake) for large-scale molecular datasets;
- Develop and orchestrate ML pipelines using tools such as Argo Workflows or Prefect;
- Implement and optimize distributed training and inference workflows using Ray / Anyscale;
- Contribute to development and deployment of predictive models within a computational drug discovery platform;
- Design and maintain LLM-based agentic systems supporting automated drug design workflows;
- Collaborate closely with ML infrastructure and research teams to ensure alignment between scientific needs and engineering standards.
- 5+ years of professional experience in Machine Learning Engineering, Data Engineering, or ML Infrastructure roles;
- Strong proficiency in Python and its scientific ecosystem (NumPy, Pandas, PyTorch);
- Proven experience building and operating production-grade ML systems in cloud environments;
- Solid understanding of MLOps practices, including model registries, feature stores, CI/CD for ML, and deployment patterns;
- Hands-on experience with workflow orchestration tools such as Argo Workflows or Prefect;
- Strong experience with cloud data warehouses (BigQuery or Snowflake), including performance tuning and complex SQL optimization;
- Experience working in cloud-native environments (preferably GCP) and Kubernetes-based infrastructure;
- Familiarity with AI-assisted development workflows / coding agents used in production environments;
- Ability to communicate effectively with both engineering teams and scientific/research stakeholders.
- Background in cheminformatics (e.g. RDKit, molecular featurization) or computational biology;
- Experience with distributed computing frameworks such as Ray / Anyscale;
- Knowledge of agentic AI systems and LLM-based automation workflows;
- Experience in scientific, research-driven or high-performance ML environments.
- Opportunity to work on exciting, international projects in cutting-edge industries like Automotive, Biotech, IoT;
- Possibility to develop in cloud technologies;
- Becoming part of a team that believes that the next step to a promising future is to put your heart into it and make it happen;
- Respect for your private life so you don't have to work overtime or on weekends;
- Team Events budget to socialize outside of work;
- Company Events to celebrate smaller and bigger successes (Summer Party, Programmer's Day, and trips abroad – so far we've been in South Africa, Are, and Barcelona).
- Fully remote work or in our office in Wrocław;
- Benefits such as Luxmed, Multisport, and life insurance in Nationale Nederlanden;
- Attractive referral system (9,5k for senior, 6k for mid, 2,5k for junior);
- Personal Training Budget with additional paid hours;
- Passion Day - an extra day off for your hobby to spend as you please;
- Flexible working hours with no micro-management approach. Our core hours are 9-15, the rest of the working time is up to you;
- We provide high-quality work equipment + 2 additional monitors and accessories.
- You will be invited to an HR Screening with our IT Recruiter.
- You will have a technical interview.
- You will meet with a client.
- UoP: 14 100 – 16 600 PLN gross/month
- B2B: 100 – 130 PLN net/hour + VAT
Create a free Caio profile to unlock more results and save your role and location preferences.
Unlock free search