AI Research Engineer

Domyn; Milano; Ibrido; EUR 50.000 - 80.000; Tempo pieno

Vantaggi offerti da questo lavoro

Learning Budget

Smart Working

Stock options

Descrizione del lavoro

Domyn in Milan is seeking an AI Research Engineer to design and optimize the infrastructure that trains and serves trillion‑parameter models. You will work at the edge of research and production, touching every layer of the stack and contributing to open‑source initiatives in Europe.

Collaborate with leading labs and deploy solutions in regulated industries. The role emphasizes high‑performance computing, Python/C++/Rust, and distributed training expertise.

Competenze

• PhD or equivalent practical experience in CS/AI.
• At least 5 years as AI research engineer or 2+ years plus PhD.
• Expertise in PyTorch, TensorFlow, and JAX; distributed training knowledge.
• Strong background in parallel computing and high‑performance systems (CUDA, compilers).
• Hands‑on with ML debugging and profiling tools (TensorBoard, W&B, Nsight).
• Python and C++/Rust for high‑performance inference.

Mansioni

• Design and optimize infrastructure for training and serving trillion‑parameter models.
• Collaborate with industry labs and ship to mission‑critical deployments.
• Touch all stack layers; contribute to open‑source projects in EU AI ecosystem.
• Prototype and optimize latency, throughput, and memory usage in production models.

Conoscenze

PhD in CS/AI

5+ years AI research

PyTorch

TensorFlow

JAX

CUDA

Python

C++/Rust

Distributed training

Formazione

PhD in CS/AI

Strumenti

CUDA

NVIDIA Nsight

Ray

DeepSpeed

TensorBoard

Weights & Biases

Descrizione del lavoro

Work somewhere with the creativity of a scale-up 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 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.

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 € 50.000 and € 80.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.

Employment terms are governed by the CCNL Commercio, the Italian National Collective Bargaining Agreement for the Commerce, Distribution and Services sector.

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 .