OpenX is seeking a talented Machine Learning Engineer with strong software engineering skills to contribute to impactful R&D projects, with a specific focus on building and maintaining machine learning pipelines. This role centers on engineering solutions, collaborating with the OpenX Data Science team to optimize and advance the ML platform, by providing robust infrastructure and support.
Who We're Looking For:
Thinks long-term and seeks sustainable, robust solutions rather than temporary fixes.
Adheres to industry best practices to ensure the delivery of high-quality results.
Can take an existing machine learning model and productionize it by creating a scalable pipeline for training and optimizing the model for serving.
Will assist in migrating model training pipelines across different technologies or transitioning model implementations to other frameworks.
Provides strategic guidance for optimizing models and enhancing model training pipelines.
Designs, develops, and delivers innovative features and functionalities.
Can effectively communicate with data scientists who are developing and experimenting with models, software engineers integrating those models into broader systems, and DevOps engineers managing serving and infrastructure, facilitating collaboration to build and deploy comprehensive end-to-end solutions.
Working with the newest technologies such as Cloud Computing (GCP)
Challenges at work that are difficult to find anywhere else!
Solving important problems in a scale
Joining a company that is growing and scaling
Flexible working hours & remote work option
Key Responsibilities:
Identify and eliminate unnecessary complexity to streamline processes and improve efficiency.
Analyze and enhance the efficiency, scalability, and stability of machine learning model training pipelines.
Develop, maintain, and monitor the infrastructure responsible for training ML models by automating pipeline-related tasks, adding performance metrics, helping troubleshoot the issues, and creating alerts.
Continuously enhance the ML platform by integrating feature stores, model repositories, data versioning, and skew detection capabilities.
Optimize models by profiling, analyzing flame graphs to identify performance bottlenecks, and applying faster computational graph operations or full model transformations such as pruning, quantization, or hardware acceleration.
Collaborate with team members to deploy, manage, and integrate machine learning models.
Required Qualifications:
7+ years of experience in a similar role and a Bachelor’s degree in Computer Science or a related field.
Extensive expertise in Python programming and software engineering best practices.
Demonstrated proficiency with machine learning frameworks such as TensorFlow, Keras, or PyTorch.
Familiarity with basic SQL for querying and data manipulation.
Hands-on knowledge of ML pipelines, particularly Kubeflow Pipelines.
Basic understanding of machine learning techniques sufficient to comprehend and utilize models developed by data scientists. Advanced modeling skills are not required.
Experience with cloud platforms and their machine learning capabilities, particularly Google Cloud Platform's Vertex AI, is highly desirable.
Excellent communication skills in both Polish and English.
A willingness to learn new tools and create proof-of-concept implementations.
Strong communication skills and the ability to collaborate effectively with team members from diverse backgrounds.
Our benefits (on employment contract):
Annual performance bonus
Tax-deductible system due to copyright protection: 75%
Private health care for you and your family (covered by OpenX)
Private life and travel insurance (Covid insurance included)
MultiKafeteria program
Training: access to the LinkedIn learning platform and Tech workshops
Holiday Allowance
Pension scheme (PPK from PZU)
Additional paid day off
Free parking lot
Access to peer to peer recognition platform
Monthly work-from-home allowance and one-time payment when you join us to help you set up your home office
We celebrate team members' important personal milestones (vouchers, gifts)