AI Lead

Ayo Semiconductor•Boston, MA

1d•Onsite

About The Position

Ayo is a venture-backed early-stage startup developing processors specialized for machine learning. The processor aims to provide significant improvements in speed and power efficiency, with the goal of challenging the GPU's dominance in AI computing. The role is for a deep AI practitioner to join as a founding team member, with hands-on experience in foundation models at scale and an understanding of distributed training across thousands of GPUs. The ideal candidate will have a full-stack understanding of machine learning architectures, enjoy optimizing algorithms, and be eager to deploy and train models directly on prototype chips. Prior hardware experience is not required.

Requirements

PhD in machine learning, representation learning, theory of computation, or a related field - or equivalent industry experience working on foundation models at scale.
Experience training models at scale - distributed training across many GPUs, working with large datasets and compute.
Has built and trained neural networks from scratch
Deep knowledge of the structure and internal operation of neural networks - including how and why they behave the way they do (e.g. interpretability or explainability work is a plus).
Excitement about applying deep AI expertise to new and novel hardware environments - you don’t need prior experience with photonics or silicon, but you want to learn.
Fluent knowledge of Python
Fluency in PyTorch (preferred), TensorFlow, JAX, or other industry-standard ML software libraries

Responsibilities

Deploy and run trained models on prototype hardware and digital twins, producing working demonstrations on our chips.
Develop and adapt algorithms to train models on novel processing environments, including our prototype hardware.
Work with hardware engineers to define and refine processor architecture based on insights learned through model training and experimentation.
Maintain a deep curiosity about what makes AI systems work - and bring that curiosity to bear on how they run on new hardware.
Support go-to-market strategy development