Senior Software Engineer - JAX AI Platform (Equity)
$184k - $356.5kNVIDIA
NVIDIA is hiring senior engineers in Santa Clara, California, to develop its AI platform and enhance performance optimizations in deep learning frameworks using JAX. The role demands strong programming skills and system design expertise, along with the ability to effectively collaborate across teams.
Ideal candidates will have over six years of experience, proficiency in C/C++ and Python, and familiarity with machine learning frameworks. The competitive base salary ranges from 184,000 to 356,500 USD, depending on level and experience.
#J-18808-LjbffrVacancy posted 12 hours ago
Similar jobs that could be interesting for youBased on the Senior Software Engineer - JAX AI Platform (Equity) in Santa Clara, CA vacancy
$184k - $287.5k
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance... ...frameworks using JAX, a tool that can differentiate... ...libraries, modular software design.Highly motivated with... ...will also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$272k - $431.25k
...JAX is the framework of choice for the most ambitious AI research, and the datacenters it... ...NVIDIA seeks a Senior Engineering Manager to define... ...performance across the platform, from single-... ...years of software engineering experience... ...be eligible for equity and benefits....PlatformEmployment EquitySeniorFull time- ...NVIDIA is seeking senior engineers to advance its AI platform, focusing on performance optimizations in deep learning frameworks using JAX. The role emphasizes modular, fast, and coordinated tooling to handle data, training and analysis for diverse DL solutions. Strong...PlatformEmployment EquitySenior
NVIDIA Gruppe
Santa Clara, CA11 hours ago $152k - $241.5k
...advancements in AI and machine... ...and motivated engineers to join our TensorRT... ...inference software for NVIDIA AI accelerators... .... As a Senior Software Engineer... ..., PyTorch, JAX.Knowledge of close... ...effective computing platform driving our... ...eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$152k - $241.5k
...performance computing platforms are powering the AI revolution across... ...industries. Within our software stack, CUTLASS... ...Computer Science, Computer Engineering, or related field (... ...like PyTorch, JAX, SGLang, vLLM, TRT-LLM... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
...accelerated computing platform is foundational to modern HPC and AI. At the center of... ...scalable GPU-accelerated software. We are hiring a Senior Software Engineer to develop the Rust... .../Python, PyTorch, JAX, Numba, CuPy, or... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
...unlimited potential of AI to define the next... ...for outstanding Senior Deep Learning Software Engineers to develop and productize... ...working on our platforms.What you'll be doing... ...frameworks (e.g. PyTorch, JAX) and compiler... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull timeWork experience placement$184k - $287.5k
NVIDIA is seeking a Software Engineer to join our Isaac Spatial... ...future of physical AI!What you'll be doing:Design... ...including Sim/Lab, Platform and SQA to deploy, validate... ...frameworks (PyTorch, JAX, TensorFlow).... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
...accelerated computing platform is foundational to modern HPC and AI. At the center of... ...scalable GPU-accelerated software.We are hiring a Senior Software Engineer to advance the... ...C++/Python, PyTorch, JAX, Numba, CuPy, or similar... ...be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
...NVIDIA's DGX Cloud AI Efficiency Team... ...AI infrastructure software engineer to join our team.... ...of AI systems.As a senior DGX Cloud AI Infrastructure... ...NVIDIA's AI platforms.Define meaningful... ...PyTorch, TensorFlow, JAX, and RayNVIDIA... ...also be eligible for equity and benefits....PlatformEmployment EquitySeniorFull timeRemote work- ...accelerated computing platform is foundational to modern HPC and AI. At the center of... ...scalable GPU‑accelerated software. We are hiring a Senior Software Engineer to advance the... ...C++/Python, PyTorch, JAX, Numba, CuPy, or similar... ...be eligible for equity and benefits. Application...PlatformEmployment EquitySenior
NVIDIA Corporation
Santa Clara, CA12 hours ago $182k - $242k
...Essential Cloud for AI™. Built for... ...delivers a platform of technology,... .... Our stack is engineered for speed, scale... ...looking for a Senior Engineer for CoreWeave... .../accelerator software, or performance... ...GPUs. JAX and its Pallas... ...alignment and internal equity when...PlatformEmployment EquitySeniorPermanent employmentTemporary workCasual workWork at officeFlexible hoursSocket.dev
Sunnyvale, CA12 hours ago- ...NVIDIA Corporation seeks a Senior Software Engineer to advance its JAX-based platform and performance optimizations in AI. You will design and implement JAX core components, collaborating with researchers to build scalable, fast tooling for data, training and analysis...PlatformSenior
NVIDIA Corporation
Santa Clara, CA12 hours ago $184k - $287.5k
...unlimited potential of AI to define the next era... ...into a unified production platform. Autonomous driving is... ...knowledge of E2E AV software integration from perception... ...See:We’re looking for engineers who can build systems —... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
...Architecture Group is seeking a senior software engineer to automate and optimize... ...performance analysis workflows for AI training and inference... ...building developer tools or platforms for ML... ...You will also be eligible for equity and benefits.Applications for...PlatformEmployment EquitySeniorFull timeRemote work$184k - $287.5k
...deep learning ignited modern AI — the next era of computing —... ...advancement.Are you a motivated system software engineer with a deep understanding of... ..., a core component of our platform for accelerating general... ...will also be eligible for equity and benefits.Applications for...PlatformEmployment EquitySeniorFull time$184k - $287.5k
We are hiring software engineers to work on the CUDA driver, a core component of our platform for accelerating general purpose computation on... ...You will also be eligible for equity and benefits.Applications for... ...existing vacancy. NVIDIA uses AI tools in its recruiting processes...PlatformEmployment EquitySeniorFull time$152k - $241.5k
...next generation of edge AI for automotive and robotics. We build the software stack that enables... ...efficiently on embedded and edge platforms — delivering cutting-... ..., Electrical/Computer Engineering, or a closely related... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time- ...RoboForce RoboForce is an AI robotics company... ...We are looking for a Senior Software Engineer to build scalable AI... ...create a high-performance platform for Physical AI.... ...pipelines leveraging PyTorch, JAX, or equivalent... ...Competitive stock options/equity programs. Health,...PlatformEmployment EquitySeniorFull timeWork at officeVisa sponsorship
RoboForce
Milpitas, CA1 day ago $152k - $241.5k
...high-performance computing platforms are powering the AI revolution across many... ...and industries. Within our software stack, CUTLASS stands out... ...Computer Science, Computer Engineering, or related field (or equivalent... ...will also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$152k - $241.5k
We are now looking for a Senior Software Engineer, Agentic AI! NVIDIA has been transforming computer graphics... ...sophisticated coding agents and agentic platforms capable of multi-step reasoning,... ...Level 4.You will also be eligible for equity and benefits.Applications for this...PlatformEmployment EquitySeniorFull time$224k - $356.5k
Our Automotive Platform Team is building the software foundation for scalable,... ...looking for exceptional engineers who thrive on... ...computing.We are seeking a Senior Software Engineer... ...platform performance, AI model optimization,... ...be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
...Group is seeking excellent software engineers to enable the next wave of NVIDIA... ...NVIDIA’s diverse set of platforms, from datacenter to autonomous... ...will also be eligible for equity and benefits.Applications for... ...vacancy. NVIDIA uses AI tools in its recruiting processes...PlatformEmployment EquitySeniorFull timeRemote work$152k - $241.5k
...into the unlimited potential of AI to define the next era of computing... ...on the world.We are seeking a Software Engineer - Scientific Evaluation to own a shared platform for classical testing,... ...4.You will also be eligible for equity and benefits.Applications for this...PlatformEmployment EquitySeniorFull time- ...Cloud, is a leader in AI cloud infrastructure serving... ...Tuesday.Fleet Engineering at Lambda is responsible... ...ties them together.As a Senior Software Engineer in Fleet... ...accelerator, and network platforms, as well as new datacenter... ...offer generous cash & equity compensationHealth,...PlatformEmployment EquitySeniorWork at officeLocal areaWork from homeFlexible hours
$136k - $218.5k
...into the unlimited potential of AI to define the next era of... ...are building a next-generation software platform for semiconductor development... ...endeavor, and we need engineers who thrive in a hands-on environment... ...will also be eligible for equity and benefits.Applications for...PlatformEmployment EquitySeniorFull time$184k - $287.5k
.... We're looking for a senior engineer at the intersection of... ...systems, and advanced data platforms. This role will help... .... As a Senior Software Engineer, you will design... ...simulation environments, AI models, and emerging... ...also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$184k - $287.5k
We are now seeking a Senior Infrastructure Software Engineer for NVIDIA TensorRT Edge-LLM!NVIDIA... ...next generation of edge AI. This is an outstanding... ...across NVIDIA’s diverse set of platforms, from Drive AGX for... ...will also be eligible for equity and benefits.Applications...PlatformEmployment EquitySeniorFull time$212.5k - $250k
...JoinCarta is the connected platform and AI-native ecosystem for... ...brings together the software, services, and legal... ...founders use to manage equity, fund managers use to... ...lifecycle so Carta's engineers can do the best work of... ...Problems You’ll SolveAs a Senior Software Engineer II...PlatformEmployment EquitySeniorFull time$152k - $241.5k
We are now looking for a Senior DFX Software Engineer - Machine Learning. Do you like... ...state-of-the-art compute platforms and Artificial Intelligence... ...will also be eligible for equity and benefits.Applications for... ...vacancy. NVIDIA uses AI tools in its recruiting processes...PlatformEmployment EquitySeniorFull time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior Software Engineer - JAX AI Platform (Equity). Be the first to apply!
Related searches
- senior robotics software engineer Santa Clara, CA
- software system engineer Santa Clara, CA
- security software engineer Santa Clara, CA
- intel software engineer Santa Clara, CA
- software developer fintech Santa Clara, CA
- new graduate software engineer Santa Clara, CA
- software development engineer aws Santa Clara, CA
- information technology software engineer Santa Clara, CA
- graduate software developer no experience Santa Clara, CA
- software developer internship no experience Santa Clara, CA