$184,000 – $287,500
Listed on NVIDIA’s own careers site. You apply with them directly — we never stand between you and the employer.
What this role is
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. The full posting is on their site.
Skills this role names
Log in to see which of these are already on your profile.
What they ask for
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