Machine Learning Engineer – Computer Vision

A leading construction technology software company within the construction and infrastructure ecosystem is expanding its Innovation Team in Budapest. We are seeking a highly skilled Machine Learning Engineer to help develop advanced computer vision solutions that address real-world challenges in the construction and infrastructure industries.

As a Machine Learning Engineer – Computer Vision, you will design and develop state-of-the-art computer vision models that convert visual and spatial data into actionable insights. Your work will focus on building robust AI systems capable of interpreting images, documents, and spatial data used across construction and infrastructure workflows.

As part of the Innovation Team, you will collaborate with engineers, researchers, and product teams to identify opportunities where computer vision can drive meaningful innovation. You will also play a key role in transforming experimental models into scalable, production-ready solutions that can be integrated into enterprise platforms.

• Design, train, and evaluate computer vision models for tasks such as detection, segmentation, classification, and image understanding.
• Utilize pre-trained models and optimize deep learning architectures for practical, real-world applications.
• Apply generative image models, including diffusion-based approaches, to enhance visual data processing capabilities.
• Convert research prototypes into reliable and scalable production systems.
• Prepare and curate datasets, conduct benchmarking, and evaluate model performance.
• Collaborate with cross-functional teams to integrate computer vision features into products and services.
• Contribute to innovation initiatives addressing challenges such as document vectorization and point cloud data processing.

• Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• Strong expertise in computer vision and deep learning.
• Hands-on experience with frameworks such as PyTorch or TensorFlow, and tools like OpenCV.
• Experience with CNN architectures and generative models, including diffusion-based methods.
• Knowledge of model optimization techniques, evaluation metrics, and MLOps practices.
• Ability to transform experimental models into reliable production solutions.
• Strong analytical and problem-solving abilities with a proactive, solution-oriented mindset.
• Excellent communication and collaboration skills.

• Almost fully remote work model, with one office day per week (office located near Allee Shopping Mall).
• Opportunity to work with cutting-edge AI technologies in computer vision and deep learning.
• Participation in a high-impact innovation team driving AI initiatives in construction technology.
• Collaborative and research-focused work environment.
• Long-term, stable employment in a full-time permanent position.
• Opportunities for professional development and involvement in pioneering AI projects.