Requirements

PhD or MS degree in Computer Science, Electrical Engineering, Computer Engineering, or related field.
Extensive experience in system-level programming and optimization.
Proficiency in designing, evaluating, and optimizing hardware systems for machine learning applications.
Strong programming skills in Python, C, and familiarity with machine learning frameworks like PyTorch.
Understanding of machine learning algorithms, data structures, and software-hardware interaction principles.
Experience with parallel programming, multithreading, synchronization.
Ability to optimize code for performance enhancements.
Strong communication skills for team collaboration.

What You'll Be Doing

Design and optimize software and hardware systems for machine learning workloads.
Collaborate with teams to develop high-performance machine learning solutions.
Develop toolchains for improved efficiency on specialized hardware.
Analyze and optimize interconnects between hardware components.
Conduct performance analysis, identify bottlenecks, and implement solutions.
Stay updated on latest advancements in machine learning technologies.