AI Research Engineer

AI Research Engineer

Domyn

Milan, Italy

As an AI Research Engineer, you'll sit at the intersection of research and production, designing and optimizing the infrastructure that trains and serves these models. Expect to touch every layer of the stack, and contribute to open-source projects shaping the European AI ecosystem.

What You Have

• PhD in Computer Science, Artificial Intelligence or a related field, or equivalent practical experience.
• At least 5 years of proven experience as an AI research engineer or more than 2 years of experience and a PhD.
• Expertise in modern machine learning frameworks such as PyTorch, TensorFlow, and JAX, with deep knowledge of distributed training (Pytorch Distributed, Ray, DeepSpeed).
• Strong background in parallel computing and high-performance systems, including CUDA programming and compiler optimizations.
• Hands-on experience with ML model debugging and performance profiling tools (TensorBoard, Weights & Biases, NVIDIA Nsight).
• Proficiency in Python and C++ or Rust, particularly for high-performance inference and AI accelerators.
• Solid understanding of mathematics behind deep learning (linear algebra, probability, optimization).
• Experience deploying models in production, optimizing for latency, throughput, and memory efficiency.

What Would Be Nice to Have

• Track record of publications at top-tier AI venues, reflecting a deep understanding of frontier research.
• Contributions to open-source AI projects, demonstrating systems-level thinking and impact in the research community
• Knowledge of AI safety, alignment, or adversarial robustness techniques.
• Experience with low-precision training (FP8, FP4), quantization, pruning, or distillation at scale.

Who you are

• A versatile engineer who is comfortable operating across both research and production environments, enjoys solving complex infrastructure challenges, and is motivated by building scalable AI systems with real-world impact.
• Unwavering Excellence, your academic or professional journey is marked by consistent accomplishments, always striving for superior outcomes.
• Driven and Fearless, motivated and unafraid of challenges, you proactively overcome obstacles.
• Passionate Enthusiast, you have a genuine love for the field, you stay updated with the latest developments.
• Systems Thinker, you thrive at the intersection of research and engineering, optimizing infrastructure for trillion-parameter models.
• Collaborative Team Player, enthusiastic, curious, and adept at problem-solving, you excel independently and in a team environment.
• Fluent in English, you're eager to contribute in a multicultural, international setting.