Junior AI Research Engineer — Learn, Build, Scale with Equity

Domyn; Milano; On-site; EUR 30,000 - 50,000; Full time

Benefits offered by this job

Learning Friday

Smart Working

Salary based on experience

Job summary

Domyn in Milan is seeking a Junior AI Research Engineer to design and optimize the infrastructure for training and serving trillion-parameter models, covering pretraining, post-training, and deployment.

You will work across research and production, touching every layer of the stack and contributing to open-source projects shaping the European AI ecosystem. The role requires a Master’s degree and up to 2 years of relevant experience, with Python and ML frameworks as core skills.

Qualifications

• Master’s Degree in Computer Science, AI or related field.
• Up to 2 years of proven experience as an AI research engineer or in similar roles.
• Hands-on experience with at least one modern ML framework (PyTorch, TensorFlow, or JAX). Familiarity with distributed training is a plus.
• Basic understanding of parallel computing and high-performance systems; exposure to CUDA or compiler optimization is a plus.
• Hands-on experience with ML model debugging tools such as TensorBoard and Weights & Biases; exposure to profiling tools like Nsight is a plus.
• Proficiency in Python and familiarity with C++ or Rust; exposure to AI inference or accelerators is a plus.
• Solid understanding of mathematics behind deep learning.
• Exposure to deploying models in production, optimizing latency, throughput, and memory.

Responsibilities

• Design and optimize the infrastructure that trains and serves large-scale models.
• Touch every layer of the stack from research to production.
• Contribute to open-source projects shaping the European AI ecosystem.

Skills

Python

PyTorch

TensorFlow

JAX

Distributed training

Math for DL

C++/Rust

English fluency

CUDA programming

TensorBoard

Weights & Biases

NVIDIA Nsight

Education

Master’s Degree in Computer Science, AI or related field

Tools

CUDA programming

TensorBoard

Weights & Biases

NVIDIA Nsight

Job description

Domyn in Milan is seeking a Junior AI Research Engineer to design and optimize the infrastructure for training and serving trillion-parameter models, covering pretraining, post-training, and deployment.

You will work across research and production, touching every layer of the stack and contributing to open-source projects shaping the European AI ecosystem. The role requires a Master’s degree and up to 2 years of relevant experience, with Python and ML frameworks as core skills.