Machine Learning Engineer

About This Opportunity

We are representing a confidential AI software company that is building advanced production scheduling solutions for the manufacturing sector. Our client combines cutting-edge machine learning with simulation environments to help factories optimize their operations. They work with manufacturers internationally and maintain strategic partnerships with enterprise systems providers.

This is a Machine Learning Engineer role based in Berlin with flexible remote arrangements. You'll work on core AI technology, developing reinforcement learning agents and the simulation environments they train in. This position offers significant ownership and the chance to apply ML to real-world manufacturing challenges.

Responsibilities

• Write robust, scalable and production-ready Python code for machine learning systems
• Provide code reviews, guidance and mentorship to maintain high coding standards
• Write unit and integration tests using tools such as unittest and pytest
• Design, engineer and optimize simulation features for reinforcement learning 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 RL agents, including algorithms, reward functions and architecture

What We're Looking For

Must Haves:

• Background in Computer Engineering, Mathematics, Machine Learning, Industrial Engineering or 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

Nice to Haves:

• 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

How We Work With You

When you apply through Get A Job.ai, our talent team will review your background and conduct an initial screening to understand your experience and goals. If there's a strong match, we'll submit your profile to our client for consideration. Please note that you should not contact the client directly—all communication will be coordinated through our recruiting team to ensure the best experience for everyone involved.

Pay

Compensation details will be discussed during the screening process based on your experience level and the client's budget for this role.

Additional Benefits from the Client:

• Ownership from day one in a small team with fast feedback cycles
• Collaborative environment working alongside ML specialists and software engineers
• Supportive, open culture with clear communication and flat hierarchies
• Flexible working hours and hybrid setup (remote or co-working space in Berlin Mitte)
• Company-provided laptop and equipment

Get A Job.ai is an equal opportunity recruiter. We welcome applications from candidates of all backgrounds and are committed to building diverse teams for our clients.