OpenReq
Software Engineer, LLM Compilation
OpenReq, Cupertino, California, United States, 95014
About EtchedEtched is building AI chips that are hard-coded for individual model architectures. Our first product (Sohu) only supports transformers, but has an order of magnitude more throughput and lower latency than a B200. With Etched ASICs, you can build products that would be impossible with GPUs, like real-time video generation models and extremely deep chain-of-thought reasoning.
Software, LLM Compilation
Software sells chips. Etched ASICs are no exception. While our first chip, Sohu, is only able to run transformer models, we still need production-grade software to map existing LLMs onto our chip.
You will help make this happen. You will write optimized kernels for the operations that make up a transformer, like attention, model parallelism, and normalization, and package them into components that developers can use (e.g. in the way that vLLM has its fused `MergedColumnParallelLinear` component). You will work with the hardware team to debug issues that hurt performance.
You will work with the software team to build integrations with existing libraries like vLLM and HuggingFace Transformers, so that our software can be drop-in compatible. You will not build a Pytorch compiler stack - instead, we will build a few highly-optimized fused kernels that can be used to implement transformer models.
Representative projects:Write an optimized kernel to compute a new attention variant on our hardwareImplement HuggingFace's `CohereForCausalLM` class using Etched's transformer building blocksImplement a synchronization mechanism to coordinate between the host CPU and Etched acceleratorImplement FP8 quantization for FP16 models using the same mechanism as TransformerEngineYou may be a good fit if you:
Have 3+ years of software engineering experienceHave experience working with machine learning operatorsAre comfortable doing low-level embedded programmingPick up slack, even if it goes outside your job descriptionAre results-oriented, and bias towards shipping productsWant to learn more about machine learning researchWe encourage you to apply even if you do not believe you meet every single qualification.
Strong candidates may also have experience with:
Transformer optimizations, such as FlashAttentionOngoing research in machine learningHow we're different:
Etched believes in the Bitter Lesson. We think most of the progress in the AI field has come from using more FLOPs to train and run models, and the best way to get more FLOPs is to build model-specific hardware. Larger and larger training runs encourage companies to consolidate around fewer model architectures, which creates a market for single-model ASICs.
We are a fully in-person team in Cupertino, and greatly value engineering skills. We do not have boundaries between engineering and research, and we expect all of our technical staff to contribute to both as needed.
Benefits:
Full medical, dental, and vision packages, with 100% of premium covered, 90% for dependentsHousing subsidy of
$2,000/month
for those living within walking distance of the officeDaily lunch and dinner in our officeRelocation support for those moving to Cupertino
Software, LLM Compilation
Software sells chips. Etched ASICs are no exception. While our first chip, Sohu, is only able to run transformer models, we still need production-grade software to map existing LLMs onto our chip.
You will help make this happen. You will write optimized kernels for the operations that make up a transformer, like attention, model parallelism, and normalization, and package them into components that developers can use (e.g. in the way that vLLM has its fused `MergedColumnParallelLinear` component). You will work with the hardware team to debug issues that hurt performance.
You will work with the software team to build integrations with existing libraries like vLLM and HuggingFace Transformers, so that our software can be drop-in compatible. You will not build a Pytorch compiler stack - instead, we will build a few highly-optimized fused kernels that can be used to implement transformer models.
Representative projects:Write an optimized kernel to compute a new attention variant on our hardwareImplement HuggingFace's `CohereForCausalLM` class using Etched's transformer building blocksImplement a synchronization mechanism to coordinate between the host CPU and Etched acceleratorImplement FP8 quantization for FP16 models using the same mechanism as TransformerEngineYou may be a good fit if you:
Have 3+ years of software engineering experienceHave experience working with machine learning operatorsAre comfortable doing low-level embedded programmingPick up slack, even if it goes outside your job descriptionAre results-oriented, and bias towards shipping productsWant to learn more about machine learning researchWe encourage you to apply even if you do not believe you meet every single qualification.
Strong candidates may also have experience with:
Transformer optimizations, such as FlashAttentionOngoing research in machine learningHow we're different:
Etched believes in the Bitter Lesson. We think most of the progress in the AI field has come from using more FLOPs to train and run models, and the best way to get more FLOPs is to build model-specific hardware. Larger and larger training runs encourage companies to consolidate around fewer model architectures, which creates a market for single-model ASICs.
We are a fully in-person team in Cupertino, and greatly value engineering skills. We do not have boundaries between engineering and research, and we expect all of our technical staff to contribute to both as needed.
Benefits:
Full medical, dental, and vision packages, with 100% of premium covered, 90% for dependentsHousing subsidy of
$2,000/month
for those living within walking distance of the officeDaily lunch and dinner in our officeRelocation support for those moving to Cupertino