Senior Software Engineer, JAX

🕒 August 4

🏄 California – Remote

🏄 California – Remote

💵 $184k - $356.5k / year

⏰ Full Time

🟠 Senior

🧑‍💻 Full-stack Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

👻 Ghost score 33%

What it means: how likely this posting is sitting open without active hiring.How to read it: 60% or higher means we think this is a ghost. Lower scores are less of a worry.How we score it: age, reposts, open roles vs team size, how fast jobs close, and whether pay is listed.Fine print: an estimate from public data, not a judgment of the company. Real jobs can score high, so treat it as a hint, not a fact.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

NVIDIA

WebsiteLinkedInAll Job Openings

10,000+ employees

Founded 1993

🏥 Healthcare

🏭 Manufacturing

🤖 Artificial Intelligence

Healthcare • Manufacturing • Artificial Intelligence

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

📋 Description

• Play meaningful role in NVIDIA's effort in contributing to JAX • 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 will increase the efficiency of teams developing AI-based systems • Bridge the gap between numerical computing, simulation and deep learning research and their applications in real world products

🎯 Requirements

• BS in Computer Science or Computer Engineering or related field (or equivalent experience) • 6+ years relevant experience • C/C++ and Python programming • Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.) • Proven ability developing customer-facing solutions, balancing feature requests and bugs • Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design • Highly motivated with excellent verbal and written communication skills • Ability to work successfully with multi-functional teams, principles and architects; coordinates effectively across organizational boundaries and geographies • Understanding of JAX, Autograd, tracing, code generation and DSL compilers and their design • Understanding of deep learning training in distributed contexts: multi-GPU, multi-node, synchronous vs asynchronous • Background with software shipping cycles (dev, deploy, release, CI) • Experience building distributed systems and services at large scale

🏖️ Benefits

• Equity • Benefits