Senior Software Engineer, JAX

🕒 5 days ago

🌐 Poland, Switzerland, +3 more countries – Remote

🇵🇱 Poland – Remote 🇨🇭 Switzerland – Remote 🇬🇧 United Kingdom – Remote 🇩🇪 Germany – Remote 🇫🇷 France – Remote

💵 zł292.5k - zł650k / year

⏰ Full Time

🟠 Senior

🧑‍💻 Full-stack Engineer

👻 Ghost score 1%

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 real-world product applications

🎯 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