Senior Engineer – AI Data Generation & Optimization Research

Company Overview
Axelera is a European, highgrowth Series B startup revolutionizing the AI landscape with our inmemory computing platform. We specialize in creating AI hardware and software optimized for highperformance inference, catering to cuttingedge use cases across highend edge computing, embodied AI, and serverside AI deployments. We are looking for passionate, innovative research engineers to join our team and help drive the future of AI.

Role Overview
We are seeking an AI Research Engineer with a strong focus on data generation, selection, and optimization. The role will be central to advancing our platform’s capabilities for generating highquality datasets, selecting the right data for training cuttingedge AI models, and optimizing data pipelines to maximize the performance and scalability of our AI solutions.

You will work closely with our AI researchers and engineers to create sustainable data processes that enable continuous improvement in our AI models, ensuring that our systems become smarter and more efficient over time. Your work will directly contribute to building a competitive edge for our products by harnessing the power of data.

Responsibilities:

• Data Generation: Develop innovative techniques for generating synthetic data or augmenting existing datasets to improve the quality, diversity, and representativeness of training data.

• Data Selection & Optimization: Design and implement data selection strategies that optimize model performance by choosing the most relevant data for training and validation. Focus on balancing the quality, diversity, and quantity of data to create more accurate models.

• Data Flywheel Platform: Build and optimize systems that create a selfsustaining data generation and selection loop, allowing the company to continuously collect, process, and refine data for better AI model performance.

• Collaboration: Work closely with crossfunctional teams, including AI researchers, hardware engineers, and software engineers to integrate AI models into the broader platform.

• Innovation: Stay uptodate with the latest research in multimodal AI, proposing and implementing new techniques to push the boundaries of whats possible in generative AI.

• Deployment & Testing: Implement best practices for model testing, deployment, and continuous improvement to ensure models scale effectively in production environments.

• Requirements:

  • Experience: Proven experience (for all levels) in developing and optimizing data pipelines for machine learning, with a focus on data generation, data augmentation, or data selection techniques.

  • Technical Skills:

    • Experience with datacentric AI approaches, including data cleaning, augmentation, and synthetic data generation.

    • Strong programming skills in languages like Python, and experience with data manipulation libraries (e.g., NumPy, Pandas) and nearest neighbour search libraries (e.g., FAISS).

    • Familiarity with machine learning frameworks such as TensorFlow, PyTorch, or JAX.

    • Knowledge of optimization techniques for largescale data pipelines, data storage, and distributed computing.

    • • Knowledge: A strong understanding of the latest advancements in AIML research, particularly in data optimization (e.g. taskspecific data selection, data pruning etc.). Strong analytical skills to identify and implement strategies that enhance data quality and improve model performance.

      • Collaboration & Communication: Ability to work in a highly collaborative, fastpaced startup environment and communicate complex technical concepts clearly.

      • Preferred Qualifications:

        • PhD or advanced degree in Computer Science, Machine Learning, AI, or related fields.

        • 5+ years of postgraduation relevant work experience.

        • Experience building or working with largescale data systems or platforms (e.g., distributed data pipelines, cloudbased data storage).

        • Familiarity with reinforcement learning or generative modeling techniques, particularly for data generation.

        • A passion for datadriven approaches to AI development, and the ability to translate theoretical research into actionable, scalable solutions.

        • Location

          This position is based in Italy & we support relocation to Bologna, Florence or Milan for talent based abroad and interested in this role.

          Why Join Us?

          • Impact: Play a key role in shaping the future of AI by developing the foundational data systems that drive innovation and optimize AI models at scale.

          • Culture: Join a highly collaborative, fastgrowing team that values innovation, continuous learning, and tackling challenging realworld problems.

          • Growth: As a Series B startup, youll have significant opportunities for personal and professional growth, with a chance to influence both product direction and technology strategy.

          • Compensation: Competitive salary, equity options, and a comprehensive benefits package.

          • At Axelera AI, we wholeheartedly embrace equal opportunity and hold diversity in the highest regard. Our steadfast commitment is to cultivate a warm and inclusive environment that empowers and celebrates every member of our team. We welcome applicants from all backgrounds to join us in shaping the future of AI.