Machine Learning Integrations Engineer

OneRail Poland Sp. z o.o. operates in the logistics technology sector, focusing on innovative ML-integrated solutions for their flagship product. The company offers a flexible hybrid work style and fosters career growth opportunities with a collaborative engineering culture.

• Minimum 3 years of professional full-stack development experience with ML integration.
• Hands-on experience serving ML models in production including serialization, serving frameworks, and inference optimizations.
• Strong proficiency in TypeScript and Python.
• Experience in microservices architecture, message brokers such as Kafka, and asynchronous frameworks.
• Solid knowledge of React including hooks, context, and state management.
• Deep understanding of REST principles, OpenAPI/Swagger, and API versioning/backward-compatibility.
• Experience with Docker and container orchestration.
• Familiarity with CI/CD pipelines using GitHub Actions, Jenkins, or GitLab CI and automated testing (unit, integration, e2e).
• Advanced proficiency in English (B2+).

• Work with ML pipeline engineers to deploy and maintain ML models (e.g., PyTorch, scikit-learn) as robust service endpoints.
• Collaborate with data scientists and product teams to productionize prototypes and extend product ML and AI capabilities.
• Define and evolve RESTful APIs with OpenAPI/Swagger, ensuring low-latency, high-throughput ML inference.
• Version, document, and maintain API contracts for internal and external consumers.
• Build microservices (Python—FastAPI/Flask or Node.js) encapsulating ML logic, data pre/post-processing, and business rules.
• Optimize service performance, scalability, and reliability (caching, async processing, batching).
• Implement and modify React components and pages to expose ML-driven functionality, managing back-end to front-end data flow.
• Containerize ML services using Docker and orchestrate deployments.
• Collaborate with cross-functional teams (data science, product, QA) to translate ML requirements into production features.
• Champion code quality, API versioning strategies, and thorough documentation.