# Senior Software Engineer, JAX

Hiring organization: [NVIDIA](https://career.thegoodapps.co/organizations/nvidia)

Canonical page: https://career.thegoodapps.co/jobs/b0230cf2-6fa2-4a21-92a6-a08a4e6d0b68

Listed on NVIDIA's own careers site. Applications go to them directly.

- Seniority: Senior
- Salary: 184000 – 287500 USD per year

## Summary

NVIDIA seeks a senior engineer to develop and optimize JAX-based components for its AI platform, working on core infrastructure that powers deep learning research and real-world applications. This role combines systems-level performance optimization with collaborative work across research and product teams to advance numerical computing tools.

_Our summary, not NVIDIA's wording._

## Skills named

C++, JAX, Python, PyTorch, scikit-learn, TensorFlow

## Required

- Bachelor's degree 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 framework internals
- Proven ability developing customer-facing solutions
- Technical foundation in CPU and GPU architectures and numeric libraries
- Excellent communication and coordination across teams

## Nice to have

- Understanding of JAX, Autograd, tracing, code generation, and DSL compilers
- Experience with distributed deep learning training across multiple GPUs and nodes
- Background with software shipping cycles and CI/CD
- Experience building distributed systems and services at scale

Apply on NVIDIA's site: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Software-Engineer--JAX_JR2004629