Working Student - ML Compiler

On-site | Part-time (20 hours per week) About Tensordyne Artificial intelligence (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! The Opportunity At Tensordyne, we are at the forefront of cutting-edge AI hardware and software solutions. To keep pace in this dynamic, interdisciplinary field, we are seeking a Working Student in ML Compiler Engineering to contribute to the development of compilers for our custom ASICs. This role offers an exciting opportunity to gain hands-on experience in full-stack compiler engineering while working alongside a team of experts in a fast-paced, innovative environment. If you are passionate about machine learning compilers, numerical programming, and performance optimization, we'd love to hear from you! As a working student, you'll gain valuable experience in a dynamic and innovative environment that thrives on curiosity and collaboration. This part-time role (20 hours per week) is based on-site at our Munich office.. The specific duties of this role will be shaped by both our business priorities and your individual interests. Your Responsibilities • Design and implement ML compiler passes and optimizations tailored for our custom ASIC architecture. • Develop intuitive tooling to elevate the developer experience of our production ML compiler. • Work closely with multi-disciplinary teams, ranging from ML to embedded system engineers. What you'll bring • A highly motivated, curious, and detail-oriented mindset with a strong desire to learn. • Passion for compilers • Solid understanding of graph theory and graph algorithms. • Proficiency in at least one systems programming language (e.g., C++, Rust). • Experience with Python and its ecosystem. • Currently enrolled in a Master's program in Electrical Engineering, Robotics, Computer Science, or a related AI field . Nice to Have • Familiarity with parallel programming frameworks such as CUDA or OpenMP . • Interest in programming language design and compiler theory. • Experience with performance optimization. Tensordyne's culture was built on the following values: • Put people first. We only succeed when our people succeed. • Ethics and integrity always; Being 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.