Ayo Semiconductor
Machine Learning Architect
Boston
Sponsorship not specifiedDetected 70 days ago
PythonFull-Stack DevelopmentAlgorithmsMachine LearningTensorFlowPyTorch
About the role
- We are seeking a deep ML practitioner to join as a founding team member - someone with hands-on experience working on or alongside a foundation model at scale, who understands what happens under the hood when splitting jobs across thousands of GPUs, and who is excited to bring that depth to novel hardware.
- They have a full-stack understanding of machine learning architectures, love to optimize algorithms across disciplinary boundaries, and will deploy and train models directly on our prototype chips to help us prove out what our processor can do - no prior hardware experience required.
Responsibilities
- Develop and adapt algorithms to train models on novel processing environments, including our prototype hardware.
- Support go-to-market strategy development
Requirements
- Experience training models at scale - distributed training across many GPUs, working with large datasets and compute.
- 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
Nice to have
- 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).
- Fluency in PyTorch (preferred), TensorFlow, JAX, or other industry-standard ML software libraries
Compensation
- Competitive salary.
Benefits
- Early-stage means real ownership, real impact, and meaningful equity.
- Equity commensurate with stage and seniority.
- Benefits package including health, dental, and vision.
- Maintain a deep curiosity about what makes machine learning systems work - and bring that curiosity to bear on how they run on new hardware.
- PhD in machine learning, representation learning, theory of computation, or a related field - or equivalent industry experience working on foundation models at scale.
Company info
- What We're Looking For:
- Fully onsite in Boston - we are a collaborative, in-person team.
Apply directly at Ayo Semiconductor →Create a free account for alerts like thisView Ayo Semiconductor immigration profile
This listing is sourced directly from Ayo Semiconductor's careers page and normalized into a canonical job model.