Machine Learning Research Engineer

  • US$180000 - US$230000 per annum
  • San Francisco, California

Pioneering 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.


  • 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!

Victor Pascoe Researcher

Apply for this role