Research Intern (f/m/d) – xLSTM & Sequence Modeling

Linz, Oberösterreich, Austria

NXAI

Welcome to NXAI, where we engage in cutting-edge research and product development at the heart of Europe, aiming to elevate AI technologies to unprecedented levels of performance.

View all jobs at NXAI

Apply now Apply later

Are you passionate about AI and eager to gain hands-on experience in a dynamic research and engineering environment? Our AI Research & Engineering team is currently offering a part-time or full-time internship (duration 3–6 months) with the possibility of continued employment. The start date is flexible, so you can begin whenever it suits you best!

As pioneers in next-generation AI architectures, NXAI’s work spans cutting-edge research and real-world applications. Our team thrives at the intersection of science and impact, and we’re looking for curious minds to join us in pushing what's possible 🚀.

As a Research Intern at NXAI, you'll dive deep into xLSTM and related architectures with the goal of building more efficient and expressive sequence models 🤖.

Your responsibilities include:

  • Exploring and implementing variations of xLSTM and novel recurrent architectures

  • Running comparative experiments with Transformer and Mamba-based models

  • Contributing to open-source repos and internal tools

  • Analyzing performance trade-offs (latency, accuracy, memory footprint)

  • Presenting findings in team discussions and internal research talks

Requirements

You may be a good fit if you:

  • are currently enrolled in a M.Sc. or Ph.D. program in Artificial Intelligence, Machine Learning, Computer Science, or a related technical field

  • have solid understanding of deep learning fundamentals, especially in sequence modeling, optimization, and neural network architectures

  • have hands-on experience with PyTorch or JAX

  • have a collaborative mindset and strong intrinsic motivation to to contribute to cutting-edge AI research and help shape real-world applications

Bonus points:

  • First practical experience through internships or contributions to ML open-source projects

  • Prior experience with xLSTM, Mamba, S4, or similar architectures

  • Knowledge of distributed training or efficient inference

Why Intern with NXAI?

  • Access to mentorship and cutting-edge research

  • Learn from some of the best minds in the field, where professional excellence is the standard

  • Opportunity to convert to a full-time role

  • Gain hands-on experience at the forefront of a fast-growing company

  • Contribute to shaping the future of AI – and make a real impact with your work

Ready to make an impact? Apply now and let's build the future of AI together! 🤖

Tell us about your projects and what drives you – we’d love to hear your story.

Apply now Apply later
Job stats:  17  2  0
Category: Research Jobs

Tags: Architecture Computer Science Deep Learning Engineering JAX Machine Learning Open Source PyTorch Research

Region: Europe
Country: Austria

More jobs like this