AI / ML Research Scientist

About

Drive cutting-edge AI research as an AI / ML Research Scientist. Design models and training methodologies leveraging unique wafer-scale hardware that surpasses conventional platforms.

This role requires a deep understanding of optimization theory and advanced AI model design. Collaborate closely with ASIC, compiler, and kernel teams, alongside external academic partners. Shape future hardware by identifying crucial computational patterns for next-gen AI workloads.

Key Responsibilities

• Design AI models from first principles on wafer-scale hardware
• Investigate computational techniques to enhance model design
• Develop strategies considering hardware optimization and memory patterns
• Publish research findings at top-tier venues
• Influence future hardware designs based on AI requirements

Requirements

• PhD in Machine Learning, Computer Science, or related fields preferred
• Proven research record in top AI venues
• Proficiency in Python and PyTorch; knowledge of C is beneficial
• Strong mathematical maturity and communication skills
• Experience in fast-paced collaborative environments

Contribute to groundbreaking AI research that will redefine standards in computational efficiency and model architecture.