AI Research Engineer

TN Italy; Italia; In loco; EUR 50.000 - 80.000; Tempo pieno

Descrizione del lavoro

Join a forward-thinking company as an AI Research Engineer in Milan, where you'll merge innovative AI research with practical applications. In this pivotal role, you'll design and optimize infrastructure for next-generation AI models, enhancing efficiency and interpretability. Collaborate with a dynamic team in a multicultural environment, leveraging your expertise in machine learning frameworks and parallel computing. Enjoy a competitive salary, flexible working arrangements, and opportunities for professional growth as you contribute to groundbreaking AI solutions in a rapidly evolving industry.

Servizi

Training Budget

Flexible Working Hours

Company Equity

Smart Working

Competenze

• 5+ years of experience as an AI Research Engineer or equivalent.
• Strong background in parallel computing and high-performance systems.
• Expertise in modern machine learning frameworks.

Mansioni

• Design and optimize infrastructure for AI model training and deployment.
• Bridge cutting-edge AI research with large-scale production.
• Optimize AI systems for efficiency and real-world applications.

Conoscenze

Machine Learning

Deep Learning

Python

C++

Rust

Parallel Computing

CUDA Programming

AI Safety

Formazione

PhD in Computer Science

Master's Degree in AI

Strumenti

PyTorch

TensorFlow

JAX

Horovod

Ray

DeepSpeed

TensorBoard

Weights & Biases

NVIDIA Nsight

Descrizione del lavoro

Work somewhere with the creativity of a scaleup and expertise of an enterprise.

We’re looking for a world-class AI Research Engineer to join our AI Research Team in Milan. You’ll play a crucial role in bridging cutting-edge AI research with large-scale production, designing and optimizing infrastructure capable of training and deploying next-generation AI models. Your work will directly impact how we scale AI systems, making them more efficient, interpretable, and aligned with real-world applications.

Requirements

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 (Horovod, 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

• Proven record of contributing to AI publications, highlighting a keen understanding of cutting-edge research.
• Contributions to open-source AI projects, demonstrating systems-level thinking and impact in the research community.
• Knowledge of AI safety and alignment, or adversarial robustness techniques.
• Experience with compiler design for AI accelerators, optimizing model execution for TPUs or edge devices.

Who you are

• 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.
• 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.

About iGenius

iGenius is a deep-tech company specialized in the development of Artificial Intelligence solutions for companies operating in highly regulated industries, including financial services, government, or heavy industry. iGenius’ main product, Unicorn, offers tailored solutions for companies looking to integrate AI safely and effectively, mainly through two proprietary Large Language Models (LLMs). Italia 10B, is a multi-language model optimized for regulated sectors and elevated computational efficiency, while Colosseum 355B, built with latest-generation NVIDIA technology, is fit for mission-critical use cases. In addition to Unicorn, iGenius’ product offer includes Crystal, an AI agent for Decision Intelligence that analyzes business data in natural language and accurately supports strategic, insight-driven decision-making. In December 2024, iGenius joined forces with NVIDIA to build Colosseum – one of the largest AI supercomputers in the world – to support the deployment of its models with unrivaled speed, performance, and efficiency.

Active in both Europe and the United States, iGenius is one of the leading AI unicorns in the European landscape, and has attracted Fortune 500 companies, including Allianz and Intesa Sanpaolo. This led Gartner to recognize iGenius as a “Cool Vendor” in the AI Core Technologies category, as well as mention the company in its Market Guide for Conversational AI.

Please review our Privacy Policy here https://bit.ly/2XAy1gj .