ML Performance Engineer

I am currently seeking a Senior ML Performance Engineer to join the team, focusing on becoming the compute platform for AGI.**$80 million in series A funding, and $300 million valuation at the end of 2024.Enhancing AI by building powerful hardware, specifically tailored for Large Language Models (LLMs).Major VCs already involvedStrong history in shipping world class AI chipsIdeal candidates will **Expertise in ML frameworks such as JAX, PyTorch, or Tensorflow.Strong Python experience.Experience with KV-Caches.Deep knowledge of the Transformer architecture.Experience with distributed computing, high performance networking, or large-scale ML systems.As an ML Performance Engineer, you will write production-grade libraries for efficient distributed training and serving, as well as build performance models and tooling to validate and guide scheduling decisions.

Created: 2025-03-06