Lead Machine Learning Engineer

Pioneering (Lead) Machine Learning Researcher

Join our pioneering team at the forefront of AGI computing. We're dedicated to crafting a comprehensive AGI compute platform, covering hardware and software from silicon to systems. If you're passionate about pushing the boundaries in systems-focused ML research, join us on this impactful journey!

Key Responsibilities:

• Spearhead the training and optimization of Large Language Models (LLMs) tailored for our advanced hardware infrastructure.
• Conduct rigorous quality evaluations to ensure optimal performance of ML workloads for AGI.
• Architect and establish distributed infrastructure for both training and inference processes.
• Provide valuable insights and guidance on hardware architecture, particularly from a Machine Learning perspective.

Requirements:

• Demonstrate excellent proficiency in software engineering skills, contributing to the development of robust ML solutions.
• Possess hands-on experience in training and fine-tuning neural networks, with a preference for expertise in Large Language Models.
• Bonus: Previous exposure to optimizing neural networks for hardware efficiency, including considerations related to FLOPs, memory bandwidth, communication bandwidth, precision, and batch sizes.

If you are ready to make a significant impact in the field of ML research and contribute to the development of AGI, we encourage you to be a part of our innovative and collaborative team. Join us as we pave the way for the future of artificial intelligence!