1 day ago
Base Salary
$184k - $357k/yr
Responsibilities
- Contribute meaningfully to NVIDIA’s JAX development efforts.
- Design and implement JAX core components and optimize performance on NVIDIA products.
- Collaborate with AI applied researchers and leaders to build future-proof models.
- Build tools that improve the efficiency of teams developing AI-based systems.
- Bridge numerical computing, simulation, and deep learning research with real-world product applications.
Requirements
- Bachelor’s degree in Computer Science, Computer Engineering, or a related field, or equivalent experience.
- 6+ years of relevant experience.
- Strong programming skills in C, C++, and Python.
- Experience with machine learning frameworks and their internals, such as PyTorch, TensorFlow, and scikit-learn.
- Ability to develop customer-facing solutions while balancing feature requests and bugs.
- Strong foundation in CPU and GPU architectures, numerical libraries, and modular software design.
- Strong verbal and written communication, planning, and cross-functional collaboration skills.
Benefits
- Eligible for equity and benefits.
Categories
About Nvidia
Nvidia designs and sells GPUs and accelerated computing platforms for data centers, AI/ML, graphics, gaming, and automotive, monetizing through hardware, software platforms (CUDA, AI frameworks), and systems like DGX and networking. Customers include cloud providers, enterprises, researchers, and OEMs. Founded in 1993 and headquartered in Santa Clara, it is a public company traded on NASDAQ under NVDA.