AI Research Engineer – Data Generation & Optimization

European Series B startup revolutionizing AI landscape with in-memory computing platform for high-performance inference.
Milan, Metropolitan City of Milan, ItalyFlorence, Metropolitan City of Florence, ItalyBologna, Metropolitan City of Bologna, Italy
$80,000 - $150,000
Machine Learning
Senior Software Engineer
Hybrid
51 - 100 Employees
5+ years of experience
AI · Enterprise SaaS

Description For AI Research Engineer – Data Generation & Optimization

Axelera AI, a European Series B startup, is revolutionizing the AI landscape with their innovative in-memory computing platform. They are seeking an AI Research Engineer specialized in model compression to join their dynamic team. This role focuses on developing cutting-edge compression techniques for Generative AI models, optimizing them for real-time inference across various environments.

The position offers a unique opportunity to work at the intersection of advanced machine learning, in-memory computing, and high-performance AI inference. The ideal candidate will be responsible for developing and implementing sophisticated model compression techniques while maintaining or improving model accuracy. They will work with state-of-the-art AI technologies and contribute to groundbreaking advancements in AI hardware and software optimization.

The role combines technical expertise with collaborative teamwork, as the engineer will work closely with AI researchers, software engineers, and hardware engineers. The company offers a competitive compensation package, including equity options, and provides significant growth opportunities within a fast-growing startup environment. They support relocation to Italian tech hubs (Bologna, Florence, or Milan) for international talent.

Axelera AI values diversity and inclusion, fostering an environment where innovation and continuous learning are paramount. This is an excellent opportunity for someone passionate about AI optimization who wants to make a significant impact on the future of AI applications, from edge computing to embodied AI systems.

Last updated 23 days ago

Responsibilities For AI Research Engineer – Data Generation & Optimization

  • Design and implement advanced model compression techniques such as pruning, quantization, weight sharing, and knowledge distillation
  • Optimize compressed models for high-throughput and low-latency inference
  • Collaborate with AI researchers, software engineers, and hardware engineers
  • Stay current with latest AI and model compression research developments
  • Implement best practices for model testing, deployment, and continuous improvement

Requirements For AI Research Engineer – Data Generation & Optimization

Python
  • Proven experience in model compression techniques
  • Expertise in deep learning frameworks (TensorFlow, PyTorch, or JAX)
  • Experience optimizing models for resource-constrained environments
  • Strong understanding of deep learning algorithms and neural networks
  • Ability to work in a collaborative, fast-paced startup environment
  • PhD or advanced degree in Computer Science, Machine Learning, AI, or related fields preferred

Benefits For AI Research Engineer – Data Generation & Optimization

Equity
Relocation Benefits
  • Competitive salary
  • Equity options
  • Benefits package
  • Relocation support
  • Growth opportunities

Interested in this job?

Jobs Related To Axelera AI AI Research Engineer – Data Generation & Optimization

AI Research Engineer – Model Compression (all seniority levels)

AI Research Engineer role focusing on model compression techniques for high-performance inference at European AI startup Axelera

AI Research Engineer – Data Generation & Optimization (all seniority levels)

AI Research Engineer position at Axelera AI focusing on data generation, selection, and optimization for cutting-edge AI models and systems.

Sr. Machine Learning Engineer, Routing and Planning

Senior Machine Learning Engineer position at Amazon focusing on AI solutions for last-mile delivery optimization and routing

Solution Engineer - IS&T AI & Data Platforms

Senior Solutions Engineer role at Apple focusing on enterprise GenAI strategy and AI/ML platforms, offering competitive compensation and benefits.

AIML - Senior ML Engineer - Siri & Information Intelligence

Senior ML Engineer position at Apple working on Siri's Geo domain team, focusing on search ranking and query understanding using deep learning models.