Junior Ai Research Engineer

Domyn S.p.A.; Italia; Hybrid; EUR 30,000 - 50,000; Full time

Benefits offered by this job

Learning budget

Flexible work policy

Wellbeing resources

Equity opportunity

Job summary

Domyn is seeking a Junior AI Research Engineer in Milan to bridge research and production for trillion-parameter model scales. You will design and optimize the infrastructure that trains and serves cutting-edge models, collaborating with leading industry labs and deploying solutions in mission-critical environments.

The role emphasizes Python, C++/Rust, and modern ML frameworks, with a focus on latency, throughput, and memory efficiency across a scalable stack. Flexible work options are offered.

Qualifications

• Master's degree in CS/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 concepts.
• Basic understanding of parallel computing; exposure to CUDA programming is a plus.
• Hands-on experience with ML debugging tools such as TensorBoard and Weights & Biases; exposure to performance profiling tools (NVIDIA Nsight) is a plus.
• Proficiency in Python and familiarity with C++ or Rust.
• Solid understanding of mathematics behind deep learning (linear algebra, probability, optimization).
• Exposure to deploying models in production, including optimizing for latency, throughput, and memory efficiency.

Skills

Python

C++

Rust

Machine Learning

Distributed Training

Math for DL

English Proficiency

Model Deployment

Education

Master's Degree in Computer Science or related field

Tools

PyTorch

TensorFlow

JAX

CUDA

TensorBoard

Weights & Biases

NVIDIA Nsight

Job description

--

Work somewhere with the creativity of a scaleup and expertise of an enterprise. Join our AI Research Team in Milan to build the next generation of large language models, across the full life cycle: pretraining on trillions of tokens, state-of-the-art post-training, and releases spanning from a few billion to hundreds of billions of parameters. You'll collaborate directly with leading industry labs like NVIDIA, and your work will ship into mission-critical deployments with some of the largest players in banking, defense, manufacturing, and the public sector. You'll have first-class access to our proprietary sovereign HPC, powered by NVIDIA Blackwell GPUs and designed for frontier-scale model training, alongside additional allocations on European HPC infrastructure. As a Junior 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

• Master's Degree in Computer Science, Artificial Intelligence or a 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 machine learning framework (e.g., PyTorch, TensorFlow, or JAX) through academic, research, or personal projects. Familiarity with distributed training concepts is a plus.
• Basic understanding of parallel computing concepts and high-performance systems. Exposure to CUDA programming, or compiler optimization is a plus.
• Hands-on experience with ML model debugging tools such as TensorBoard and Weights & Biases. Exposure to performance profiling tools (e.g., NVIDIA Nsight) is a plus.
• Proficiency in Python and familiarity with C++ or Rust. Exposure to AI inference, model optimization, or AI accelerators is a plus.
• Solid understanding of mathematics behind deep learning (linear algebra, probability, optimization).
• Exposure to deploying models in production, including 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.
• Familiarity with 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.

What we offer

• Compensation: We offer a competitive base salary, as well as the opportunity to receive a variable bonus and company equity. The typical base salary for this role ranges between €30.000 and €50.000, based on experience. As you gain experience and make more significant contributions to the business, your compensation will be reviewed to match your impact. Employment terms are governed by the CCNL Commercio, the Italian National Collective Bargaining Agreement for the Commerce, Distribution and Services sector.
• Learning: When our people grow, we grow with them. That’s why we give every team member a dedicated learning budget to invest in books, courses, conferences, or anything else that fuels their curiosity and supports their role.
• Flexible working: We offer a flexible work policy that lets employees choose to work from home or the office, whatever suits them best.
• Wellbeing: We provide resources to support the mental wellbeing of our team members.

Why work at Domyn?

Ambition, talent and teamwork-this is how we shape your future, combining a start-up mindset with the impact of a corporation. Domyn is an equal opportunity employer. We welcome applicants from all backgrounds and are committed to building a diverse and inclusive team. We welcome applicants of all backgrounds, including people with visible and non-visible disabilities, and we're committed to an accessible, bias-free hiring process for everyone. Domyn is committed to accommodating applicants with disabilities. Please let us know at [e-mail]. if you need accommodation during the application or interview process. All requests are kept confidential.

About Domyn

Domyn is a frontier AI company building sovereign AI for regulated industries. Our full-stack AI operating system brings together proprietary models, enterprise knowledge, industry-specific agents, governance, and infrastructure, giving organizations ownership and control of the intelligence they increasingly rely on. We work with some of the world’s leading banks, enterprises, and public institutions to build AI they can adapt, govern, and trust in mission-critical environments. Founded in Milan in 2016, Domyn brings together researchers, engineers, and industry experts across Europe, the United States, and India. We are united by the ambition to advance the frontier of AI and apply it to some of the world’s most important and complex challenges. At Domyn, people are trusted to think independently, move with purpose, and take real ownership of their work. Please review our Privacy Policy here https://bit.ly/4tndszN.

--