Machine Learning Engineer

As a Machine Learning Engineer, you'll work on the core AI behind our production scheduling software. You'll develop and improve our reinforcement learning agents, work on the Digital Twin environments they train in and help turn complex manufacturing problems into models that work in real factories.

You'll have a lot of ownership over how we approach these problems and work closely with both our engineering team and customers. We're looking for someone who brings strong Python experience together with an understanding of manufacturing environments and wants to apply reinforcement learning to real-world production challenges.

• Write robust, scalable and production-ready Python code

• Provide code reviews, guidance, and mentorship to fellow developers to maintain high coding standards

• Write unit and integration tests using tools such as unittest and pytest

• Design, engineer and optimize features in the Digital Twin for reinforcement learning simulations using Python, NumPy and Pandas

• Create, optimize and maintain training and evaluation scripts for RL agents

• Set up and maintain Python environments using modern tools such as uv and conda

• Work collaboratively using git (GitHub)

• Participate in customer calls to understand requirements and translate them into actionable technical features

• Brainstorm and develop ideas to improve the RL agent, including algorithms, reward functions and architecture

• A background in Computer Engineering, Mathematics, Machine Learning, Industrial Engineering or a related field

• 5+ years of experience with Python

• 2+ years of experience in factory shopfloor operations as an engineer or planner, or 2+ years of experience working with ERP/MES systems for factories

• Strong understanding of manufacturing processes across different production environments, including discrete manufacturing, line production and engineer-to-order

• Basic understanding of reinforcement learning and experience developing or applying RL algorithms

• 3+ years of experience in factory shopfloor operations as an engineer or planner, or 3+ years of experience working with ERP/MES systems for factories

• Research experience in developing RL algorithms

• Experience with CI/CD pipelines, particularly GitHub Actions

• Experience with libraries and tools such as PyTorch, Optuna and MLflow