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.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.
Tags: Architecture Computer Science Deep Learning Engineering JAX Machine Learning Open Source PyTorch Research
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.