[Hiring] Working Student - LLM Inference Performance @Tensordyne

Role Description

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

• Write and optimize low-level kernels.
• Benchmark performance on real silicon.
• Develop profiling tools to help engineers understand and improve LLM inference performance.
• Gain hands-on experience with our Rust programming stack and Python SDK.
• Use AI coding agents as part of your development workflow.
• Collaborate closely with engineers working on compilers, runtimes, and kernel optimization.
• Take 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.

Qualifications

• 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.

Benefits

• 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.

Company Description

Tensordyne is an AI system solution company that builds very high-performance, low-power generative AI inference systems. We are at the leading edge of advancing the latest research and product improvements for generative AI inference solutions.