Senior Software Engineer, JAX

🕒 Agosto 4

🏄 California – Remoto

🏄 California – Remote

💵 $184.000 - $356.500 / ano

⏰ Tempo Integral

🟠 Sênior

🧑‍💻 Engenheiro Full-stack

🦅 Patrocina Visto H1B

Esta empresa já patrocinou vistos H1B no passado.

Candidatar-se

Encontrar Vagas Remotas Similares

Configurar Alerta de Email para Novas Vagas

📊 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

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

• Contribute to NVIDIA's JAX efforts • Design and implement JAX core components and drive peak performance on NVIDIA products • Work with AI applied researchers and leaders to build future-proof models • Build tools that increase the efficiency of teams developing AI-based systems • Bridge numerical computing, simulation, and deep learning research with real-world product applications

🎯 Requisitos

• BS in Computer Science, Computer Engineering, or related field, or equivalent experience • 6+ years of relevant experience • C/C++ and Python programming • Experience with machine learning frameworks and their internals, such as PyTorch, TensorFlow, or scikit-learn • Proven ability to develop customer-facing solutions while balancing feature requests and bugs • Technical foundation in CPU and GPU architectures, numeric libraries, and modular software design • Excellent verbal and written communication skills • Ability to work successfully with multifunctional teams, principals, and architects • Ability to coordinate effectively across organizational boundaries and geographies • Understanding of JAX, Autograd, tracing, code generation, and DSL compiler design • Understanding of distributed deep learning training, including multi-GPU, multi-node, and synchronous versus asynchronous approaches • Background with software shipping cycles, including development, deployment, release, and CI • Experience building distributed systems and services at large scale

🏖️ Benefícios

• Equity • Benefits • Inclusive work environment • Equal opportunity employer