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

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.

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. 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. Learning Friday Smart Working Salary based on experience