Senior Staff Engineer, AI HW/SW Codesign - California, MO, United States - Conductor

Description

What You'll Do

The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads while minimizing energy consumption and maximizing performance. To achieve this goal, we collaborate closely with both hardware and software engineers to identify and address the unique challenges posed by AI/ML workloads and to explore new computing abstractions that can provide a better balance between the hardware and software components of our systems. Additionally, we continuously conduct research and development in emerging technologies and trends across memory, computing, interconnect, and AI/ML, ensuring that our platforms are always equipped to handle the most demanding workloads of the future. By working together as a dedicated and passionate team, we aim to revolutionize the way AI/ML applications are deployed and executed, ultimately contributing to the advancement of AGI in an affordable and sustainable manner. Join us in our passion to shape the future of computing

Location: Hybrid, working onsite at our office 3 days per week with the flexibility to work remotely the remainder of your time

Reports to: SVP

• Track ML trends and identify emerging technologies that can be leveraged to improve the performance and efficiency of deep learning applications.
• Analyze workloads to identify bottlenecks and opportunities for optimization in both hardware and software components.
• Suggest new hardware innovation ideas that address identified bottlenecks and opportunities for optimization.
• Work closely with hardware architects and software engineers to develop and refine hardware and software solutions that implement the suggested innovation ideas.
• Collaborate with cross-functional teams to define and deliver ML hardware and software features and improvements.
• Troubleshoot and debug hardware and software issues, and provide technical support to customers.
• Stay up-to-date with the latest trends and advancements in the field of ML hardware and software.

What You Bring

• BS in Computer/Electrical Engineering or Computer Science with 15+ years of working experiences in silicon development or MS in Computer/Electrical Engineering or Computer Science with 13+ years of relevant working experience or PhD and 10+ years of relevant working experience preferred.
• Strong knowledge of ML workload, ML software infrastructure, compiler, and hardware.
• Experience with deep learning frameworks such as PyTorch, Tensorflow, or JAX.
• Familiarity with hardware architectures such as CPUs, GPUs, TPUs, and NPUs.
• Strong analytical and problem-solving skills
• Excellent communication and interpersonal skills
• Ability to work independently and as part of a team
• You're inclusive, adapting your style to the situation and diverse global norms of our people.
• An avid learner, you approach challenges with curiosity and resilience, seeking data to help build understanding.
• You're collaborative, building relationships, humbly offering support and openly welcoming approaches.
• Innovative and creative, you proactively explore new ideas and adapt quickly to change.

#LI-MD1