Machine Learning Engineer, AI Compiler

Overview

Role involves developing and optimizing model training pipelines in a machine learning graph compiler.

Ideal candidate should have a strong understanding of neural networks and experience with model training.

Locations

• Serbia, Central Banat, Belgrade

Requirements

• Degree in Computer Science or related field
• Strong understanding of neural networks
• Experience with model training
• Proficiency in Python and C++

Responsibilities

• Develop and optimize model training pipeline
• Benchmark and optimize performance
• Scale distributed model training
• Participate in hardware/software co-design
• Develop performance analysis infrastructure
• Integrate software into ML frameworks