Senior Software Engineer, JAX

🕒 il y a 3 jours

🏄 California – Distant

🏄 California – Remote

💵 $184 000 - $356 500 / an

⏰ Temps Plein

🟠 Senior

🧑‍💻 Développeur Full-Stack

🦅 Parrain de Visa H1B

Cette entreprise a parrainé des visas H1B par le passé.

Postuler Maintenant

Trouver des Emplois à Distance Similaires

Configurer une alerte email pour les nouveaux emplois

📊 Vérifiez votre score de CV pour ce poste

Améliorez vos chances d'obtenir un entretien en vérifiant votre score de CV avant de postuler.

Télécharger le CV

NVIDIA

Site WebLinkedInTous les Emplois

10 000+ employés

Fondée en 1993

🏥 Santé

🏭 Fabrication

🤖 Intelligence artificielle

Healthcare • Manufacturing • Artificial Intelligence

NVIDIA est une entreprise technologique de premier plan, spécialisée dans le calcul accéléré et l’intelligence artificielle (IA). NVIDIA est à l’avant‑garde des avancées en GPU (processeurs graphiques), cloud computing, centres de données et réalité virtuelle, avec un accent particulier sur les secteurs du gaming, de l’automobile, de la santé et de la robotique. Ses innovations, telles que NVIDIA Omniverse, transforment les processus numériques traditionnels en permettant des simulations haute fidélité et des tâches de rendu de pointe. Ses applications couvrent de nombreux secteurs, des véhicules autonomes avec NVIDIA DRIVE aux solutions de santé avec NVIDIA Clara, ainsi que des analyses et workflows pilotés par l’IA.

Description

• 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

🎯 Exigences

• 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

🏖️ Avantages

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