Junior AI Research Engineer

Domyn; Milano; In loco; EUR 30.000 - 50.000; Tempo pieno

Vantaggi offerti da questo lavoro

Learning Friday

Smart Working

Stock options

Company equity

Descrizione del lavoro

Domyn in Milan is seeking a Junior AI Research Engineer to bridge research and production for trillion-parameter models. You will help design and optimize the infrastructure powering pretraining, post-training, and production deployments, collaborating with leading industry labs and major enterprise users.

The role emphasizes hands-on ML framework work (PyTorch, TensorFlow, or JAX), Python proficiency, and exposure to high-performance computing and CUDA tooling.

Competenze

• 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 machine learning framework (e.g., PyTorch, TensorFlow, or JAX).
• 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.

Mansioni

• Design and optimize the infrastructure that trains and serves trillion-parameter models.
• Touch every layer of the stack, from hardware to software, balancing research and production needs.
• Collaborate across research and production teams and contribute to open-source projects shaping the European AI ecosystem.
• Help deploy models in production with a focus on latency, throughput, and memory efficiency.

Conoscenze

PyTorch

TensorFlow

JAX

Python

C++

Rust

CUDA

TensorBoard

Weights & Biases

Distributed training

Formazione

Master’s Degree in Computer Science, Artificial Intelligence or a related field

Strumenti

NVIDIA Nsight

CUDA toolkit

Descrizione del lavoro

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, includingoptimizing 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.
• Unwavening 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.

Benefits

Perks

• Learning Friday. If our team members know more, so do we. That’s why we give everyone a training budget that they can spend on books, online courses or other training materials.
• Smart Working. Trains can be a drag, you can save some commuting time by working from home.
• Salary is based on experience and topped up with other bonuses.

We offer a competitive salary, as well as an opportunity to receive company equity. The typical salary for this role ranges between € 30.000 and € 50.000. As you gain experience and make more significant contributions to the business, your compensation will be reviewed to match your impact. Additionally, depending on your seniority and your performance, you'll have the opportunity to receive stock options, with a variable value calculated from your base salary, giving you the chance to directly participate in the company’s success.

About Domyn

Domyn is a company specializing in the research and development of Responsible AI for regulated industries, including financial services, government, and heavy industry. It supports enterprises with proprietary, fully governable solutions based on a composable AI architecture — including LLMs, AI agents, and one of the world's largest supercomputers. At the core of Domyn's product offer is a chip‑to‑frontend architecture that allows organizations to control the entire AI stack — from hardware to application — ensuring isolation, security, and governance throughout the AI lifecycle. Its foundational LLMs, Domyn Large and Domyn Small, are designed for advanced reasoning and optimized to understand each business's specific language, logic, and context. Provided under an open‑enterprise license, these models can be fully transferred and owned by clients. Once deployed, they enable customizable agents that operate on proprietary data to solve complex, domain‑specific problems. All solutions are managed via a unified platform with native tools for access management, traceability, and security. Powering it all, Colosseum — a supercomputer in development using NVIDIA Grace Blackwell Superchips — will train next‑gen models exceeding 1T parameters. Domyn partners with Microsoft, NVIDIA, and G42. Clients include Allianz, Intesa Sanpaolo, and Fincantieri.

Please review our Privacy Policy here https://bit.ly/4tndszN .