Engenheiro(a) de Compilador ML, Interface de Kernel Neuron

Aceleradores AWS Machine Learning estão na vanguarda da inovação AWS. O Trainium oferece desempenho de treinamento de ML de ponta com o maior número de teraflops (TFLOPS) de poder computacional para ML na nuvem. Tudo isso é habilitado pelo AWS Neuron Software Development Kit (SDK), que inclui um compilador de ML, o compilador Neuron Kernel Interface (NKI) e um runtime que se integra nativamente a frameworks populares de ML como PyTorch e JAX.

Neuron Kernel Interface (NKI) é uma linguagem bare-metal e um compilador para programar diretamente instâncias AWS Trainium. Você pode usar o NKI para desenvolver, otimizar e executar novos operadores diretamente no hardware, aproveitando ao máximo os recursos de computação e memória disponíveis.

Explore o NKI: - awsdocs-neuron.readthedocs-hosted.com/en/latest/nki/index.html

AWS Neuron é usado em escala por clientes como Epic Games, Snap, Airbnb, Autodesk, Amazon Alexa e Amazon Rekognition, entre muitos outros em diversos segmentos.

Saiba mais sobre nossa história: - amazon.science/how-silicon-innovation-became-the-secret-sauce-behind-awss-success

Explore o Produto: - aws.amazon.com/machine-learning/neuron/

Como Engenheiro(a) de Compilador ML na equipe Annapurna Neuron, você desenvolve e escala compiladores para lidar com as maiores cargas de trabalho de ML do mundo no AWS Trainium. Você projeta e entrega recursos em toda a pilha do compilador, trabalha diretamente com equipes parceiras de ML da Amazon para desbloquear suas cargas de trabalho e aprimora suas habilidades ao lado de engenheiros seniores. Você quer trabalhar perto do hardware, ajudar a trazer novos produtos e recursos para o mercado e entregar código de alta qualidade.

Você começa o dia revisando o código de um colega para um novo recurso do compilador, deixando feedback antes que ele seja mesclado. No meio da manhã, você trabalha com uma equipe de modelos cujo kernel não está atingindo o desempenho máximo no Trainium, e você rastreia o gargalo através dos passes do compilador. Após o almoço, você escreve e testa o recurso que projetou na semana passada, e então o executa no hardware para confirmar a correção. A equipe se move rapidamente e você está sempre nos detalhes.

Sobre a equipe

Somos as equipes de compiladores por trás do AWS Neuron, o software que executa modelos de machine learning no Amazon Trainium. Nosso compilador de ML pega um modelo completo de PyTorch ou JAX e compila o grafo para ser executado em todo o chip. O Neuron Kernel Interface (NKI) é uma linguagem bare-metal e um compilador que permite aos clientes programar o Trainium diretamente, até o nível de computação e memória, para um novo operador ou um kernel mais rápido. Juntos, possuímos o caminho do modelo do framework ao silício. Nossa missão é tornar o Trainium rápido e totalmente programável.

Nossas equipes são pequenas e ágeis, e há oportunidade de aprender o tempo todo. Trabalhamos em estreita colaboração com as equipes de hardware, runtime e framework, e com as equipes de ML da Amazon que usam nossos compiladores.

Um histórico em Machine Learning, aceleradores de IA ou desempenho de Kernel é útil, mas não obrigatório.

Para ser considerado(a) para esta função, os candidatos devem residir perto ou estar dispostos a se mudar para Seattle ou Cupertino.

Qualificações Básicas:

Qualificações Preferenciais:

A Amazon é um empregador que oferece igualdade de oportunidades e não discrimina com base no status de veterano protegido, deficiência ou outro status legalmente protegido.

Candidatos do Condado de Los Angeles: As responsabilidades do cargo para esta posição incluem: trabalhar de forma segura e cooperativa com outros funcionários, supervisores e equipe; aderir aos padrões de excelência apesar das condições estressantes; comunicar-se de forma eficaz e respeitosa com funcionários, supervisores e equipe para garantir um atendimento excepcional ao cliente; e seguir todas as leis federais, estaduais e locais e as políticas da Empresa. O histórico criminal pode ter uma relação direta, adversa e negativa com algumas das principais responsabilidades deste cargo. Isso inclui os deveres e responsabilidades listados acima, bem como a capacidade de aderir às políticas da empresa, exercer bom senso, gerenciar eficazmente o estresse e trabalhar com segurança e respeito com os outros, demonstrar confiabilidade e profissionalismo, e proteger as operações comerciais e a reputação da Empresa. De acordo com o Decreto de Oportunidade Justa do Condado de Los Angeles, consideraremos para emprego candidatos qualificados com registros de prisão e condenação.

Nossa cultura inclusiva capacita os Amazonianos a entregar os melhores resultados para nossos clientes. Se você tem uma deficiência e precisa de acomodação ou ajuste no local de trabalho durante o processo de candidatura e contratação, incluindo suporte para o processo de entrevista ou integração, visite amazon.jobs/content/en/how-we-hire/accommodations para mais informações. Se o país/região em que você está se candidatando não estiver listado, entre em contato com seu Parceiro de Recrutamento.

EUA, CA, Cupertino - 165,200.00 - 223,600.00 USD anualmente

EUA, WA, Seattle - 143,700.00 - 194,400.00 USD anualmente

A faixa salarial base para esta posição é apresentada abaixo. Seu pacote Amazon incluirá pagamentos de contratação e unidades de ações restritas (RSUs). A remuneração final será determinada com base em fatores como experiência, qualificações e localização. A Amazon também oferece benefícios abrangentes, incluindo seguro saúde (médico, odontológico, oftalmológico, prescrição, seguro de vida básico e AD&D e opção de planos de vida suplementares, EAP, Suporte de Saúde Mental, Linha de Aconselhamento Médico, Contas de Despesas Flexíveis, cobertura de reembolso de adoção e barriga de aluguel), correspondência 401(k), folga remunerada e licença parental. Saiba mais sobre nossos benefícios em amazon.jobs/en/benefits.