About Us:
Join us as a Lead ML Engineer on the mission to be the leading computing platform for Artificial General Intelligence (AGI). By pushing the boundaries of hardware capabilities, we aim to elevate AI to new heights—making it better, faster, and more cost-effective. Unlike traditional approaches, we focus exclusively on Large Language Models (LLMs), streamlining our hardware and software for unparalleled simplicity.
As an Lead ML Engineer, you will contribute to the forefront of AGI research by shaping the computing platform for the future. While working in an environment that values deep domain expertise, innovative ideas, and meticulous engineering.
Key Responsibilities:
- Leverage your expertise to train and optimize Large Language Models for our cutting-edge hardware
- Conduct thorough research to ensure the optimal performance of our AI models.
- Build and set up distributed infrastructure for training and inference, contributing to the scalability of our systems.
- Provide valuable insights and advice on the hardware architecture from an ML perspective, influencing the evolution of our technology.
Requirements:
- Showcase your proficiency in software engineering, contributing to the development of robust and scalable ML systems.
- Demonstrate hands-on experience in training and fine-tuning neural networks, with a preference for expertise in Large Language Models.
- Perhaps you have experience optimizing neural networks for hardware efficiency, considering factors such as FLOPs, memory bandwidth, communication bandwidth, precision, parallel layout, and batch sizes.
Keywords: LLMs, Large Language Models, Neural Networks, Machine Learning, ML Research