AI Research Engineer

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.

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.

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

iGenius is a high-growth scaleup that thinks like an enterprise, where talented innovators can thrive and people come first. Our vision is to disrupt the B2B data industry by bringing a consumer approach to it. At iGenius, we envision a world where everyone has access to better data intelligence. We are the creators of crystal, the first virtual advisor for data intelligence.Founded in 2016 by Uljan Sharka, iGenius was selected by LinkedIn as one of Italy's Top 10 Startups in 2020. It was also mentioned in Gartner’s 2020 Hype Cycle for Analytics and Business Intelligence as an “innovative specialist vendor”, and as one of Europe’s top 100 SaaS companies, and one of nine top names in data & analytics, by venture capital firm Accel.

View all jobs
Get hired quicker

Be the first to apply. Receive an email whenever similar jobs are posted.

Ace your job interview

Understand the required skills and qualifications, anticipate the questions you may be asked, and study well-prepared answers using our sample responses.

Research Engineer Q&A's
Report this job
Apply for this job