Working Student - LLM Inference Performance

About Tensordyne

AI is transforming our world. It can perform cognitive functions that previously only humans could do, such as perceiving interactions across different modalities and environments - with the ability to quickly learn and then solve complex problems. Tensordyne is an AI system solution company that builds very high-performance, low-power generative AI inference systems. Our mission, through the creation of custom silicon, hardware and software, is to enable multimodal Generative AI inference acceleration at scale, with safe, sustainable, high-performance systems for our hyperscaler and neocloud data center customers. We are at the leading edge of advancing the latest research and product improvements for generative Al inference solutions that will make Al even more advantageous for compelling new generative AI applications.Tensordyne is a well funded, fast-paced startup company with headquarters in both Sunnyvale, CA, and Munich, Germany. We also have many talented team members working remotely across North America and Europe. We take care of our people and their families with comprehensive benefits, competitive compensation, flexible spending options, and recognition programs, because building category-defining technology starts with a healthy, supported team. Come join us as we shape the future of multimodal generative artificial intelligence!

About the Role

How fast should an LLM building block run on our hardware? How fast does it actually run, and what would make it faster?

As a Working Student, LLM Inference Performance, you'll help answer these questions by working directly with our hardware and software engineering teams.

You'll write and optimize low-level kernels, benchmark performance on real silicon, and develop profiling tools that help engineers understand and improve LLM inference performance.

You'll gain hands-on experience with our Rust programming stack and Python SDK, use AI coding agents as part of your development workflow, and collaborate closely with engineers working on compilers, runtimes, and kernel optimization.

This is an opportunity to apply your academic knowledge to real-world AI hardware challenges while taking ownership of meaningful technical projects.

What You'll Do

• Understand our hardware: Learn how Tensordyne silicon processes data, manages memory, and synchronizes operations, and how key LLM components such as attention, MLP, and Mixture-of-Experts (MoE) routing map onto the architecture..

• Develop and optimize kernels: Implement, test, and tune low-level kernels to improve the performance of LLM inference workloads..

• Analyze performance: Build analytical and roofline performance models, compare predictions against simulator and real-silicon measurements, and investigate performance bottlenecks..

• Build benchmarking and profiling tools: Develop benchmark harnesses, profiling scripts, and trace visualizations that make performance easier to measure, understand, and improve..

• Work with AI coding agents: Explore workflows where AI agents assist with kernel development, optimization, and benchmarking, and identify the tools and context they need to work effectively..

• Collaborate across engineering teams: Work closely with compiler, runtime, and kernel engineers to translate performance insights into practical improvements..

Example project: Develop a benchmarking and profiling harness for an LLM building block, such as decode attention, at two abstraction levels. Compare measured performance against theoretical estimates and make the tooling accessible to AI coding agents.

What We're Looking For

• Currently enrolled in a Master's degree program in Computer Science, Electrical Engineering, Mathematics, or a related technical field..

• Strong programming skills in Python and at least one of C, C++, or Rust..

• Understanding of computer architecture, memory systems, bandwidth, and parallel computing through academic coursework or personal projects..

• Interest in low-level programming, hardware performance, and optimization..

• Practical experience using AI coding agents in development workflows, with the ability to evaluate and improve their output..

• An experimental mindset, with the ability to form hypotheses, analyze measurements, and communicate technical findings clearly..

• Curiosity, initiative, and enthusiasm for learning new technologies..

Nice to Have

Experience in one or more of the following areas would be a plus:

• GPU or AI accelerator programming..

• Embedded systems or bare-metal programming..

• Performance profiling, benchmarking, or optimization..

• LLM inference fundamentals and architectures..

Why Join Tensordyne?

• Work on cutting-edge AI technology: Contribute to the development of next-generation AI inference hardware and software..

• Gain hands-on experience: Apply your university knowledge to real silicon and challenging engineering problems..

• Own meaningful projects: Build tools and solutions that directly support our engineering teams..

• Learn from experienced engineers: Work alongside specialists in hardware architecture, compilers, runtimes, and performance optimization..

• Explore AI-assisted development: Gain practical experience building with AI coding agents and shaping how they interact with advanced hardware systems..

• Make a lasting impact: Help develop performance tools and workflows that will support both current and future generations of Tensordyne technology..

Tensordyne's culture was built on the following values

• Put people first. We only succeed when our people succeed.
• Ethics and integrity always. Be open, honest, and respectful of everyone.
• Think Big. Be ambitious and have audacious goals.
• Aim for excellence. Quality and excellence count in everything we do.
• Own it and get it done. Results matter!
• Make each person better together than they would be as an individual.
• Embrace each other's differences, and embrace that there will be differences.

Tensordyne is an equal opportunity employer. We believe diverse teams are better equipped to solve complex problems and build exceptional technology. All qualified applicants will receive consideration for employment without regard to age, color, gender identity or expression, marital status, national origin, disability, protected veteran status, race, religion, pregnancy, sexual orientation, or any other characteristic protected by applicable laws, regulations, and ordinances.

A note to recruitment agencies: Please do not contact Tensordyne employees or leaders regarding this role. We do not accept unsolicited agency resumes and are not responsible for fees associated with unsolicited submissions.

Find more English Speaking Jobs in Germany on Arbeitnow