Senior Compiler Engineer – AI

🕒 Ontem

🏄 California, Washington – Remoto

🏄 California – Remote ☕ Washington – Remote

💵 $184.000 - $287.500 / ano

⏰ Tempo Integral

🟠 Sênior

🤖 Engenheiro de IA

🦅 Patrocina Visto H1B

Esta empresa já patrocinou vistos H1B no passado.

👻 Score fantasma 1%

O que significa: a chance de ninguém estar realmente contratando para esta vaga.Como ler: 60% ou mais significa que achamos que é uma vaga fantasma. Pontuações mais baixas preocupam menos.Como pontuamos: tempo em aberto, republicações, vagas abertas vs tamanho do time, rapidez para fechar vagas, e se o salário está divulgado.Observação: é uma estimativa de dados públicos, não um julgamento da empresa. Vagas reais podem ter pontuação alta, então use como uma dica, não como um fato.

Candidatar-se

Encontrar Vagas Remotas Similares

NVIDIA

SiteLinkedInTodas as Vagas

10.000+ funcionários

Fundada em 1993

🏥 Saúde

🏭 Manufatura

🤖 Inteligência Artificial

Healthcare • Manufacturing • Artificial Intelligence

A NVIDIA é uma empresa de tecnologia líder, especializada em computação acelerada e inteligência artificial. A companhia é pioneira em avanços em unidades de processamento gráfico (GPUs), computação em nuvem, data centers e realidade virtual, com foco nos setores de games, automotivo, saúde e robótica. As inovações da empresa, como o NVIDIA Omniverse, transformam processos digitais tradicionais ao viabilizar simulações de alta fidelidade e tarefas de renderização. Suas aplicações abrangem diversos setores, desde veículos autônomos com o NVIDIA DRIVE até soluções de saúde com o NVIDIA Clara, além de análises e fluxos de trabalho impulsionados por IA.

Descrição

• Design and implement program transformations on JAX programs (jaxpr) for large-scale training, including pipeline schedules, task placement, sharding, and communication/computation overlap • Develop optimizing compiler and runtime technologies for accelerating deep learning training on large-scale distributed systems • Compile graph intermediate representations to target-specific code • Build distributed runtimes that dispatch per-stage programs to many GPUs and move data between stages using communication libraries such as NCCL • Advance compiler technology for high-performance large-scale training workloads, including mixture-of-experts models • Collaborate with JAX, XLA, and hardware architecture teams • Integrate JaxPP into JAX training frameworks, including MaxText • Contribute to the open-source JaxPP project through maintenance, testing, performance tracking, and adoption support

🎯 Requisitos

• Master's or PhD in Computer Science, Computer Engineering, or related field, or equivalent experience • 6+ years of relevant work or research experience in compilers, distributed systems, or deep learning training • Compiler design and implementation • Distributed system design, development, and deployment • Experience with deep learning frameworks (JAX, PyTorch) • GPU training and inference of deep learning models • C/C++ and Python programming • Excellent software design skills, including debugging, performance analysis, and test design • Experience with parallel computing libraries (MPI) • Ability to work independently, define project goals and scope, and lead development efforts • Strong communication skills • Experience with JAX internals, XLA, StableHLO, or MLIR is advantageous • Experience building or tuning large training jobs with pipeline, tensor, expert, and data parallelism is advantageous • Knowledge of GPU communication libraries such as NCCL, NVSHMEM, or RDMA, and CPU and/or GPU architecture is advantageous • CUDA programming experience is a plus • Contributions to open-source compiler or machine learning projects are advantageous • Experience mentoring early career engineers and interns is a bonus

🏖️ Benefícios

• Equity • Benefits package • Competitive salary

📊 Verifique sua pontuação de currículo para esta vaga

Melhore suas chances de conseguir uma entrevista verificando sua pontuação de currículo antes de se candidatar.

Enviar Currículo