> Markdown version of [/jobs/ext/2284341-senior-software-engineer-jax](https://www.wearedevelopers.com/jobs/ext/2284341-senior-software-engineer-jax). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Senior Software Engineer, JAX

- **Company:** NVIDIA Corporation
- **Location:** Santa Clara, CA, United States
- **Experience:** Expert
- **Salary:** $184,000.0 - $287,500.0
- **Contract:** Permanent contract
- **Skills:** C++ (Programming Language), Code Generation, Computer Engineering, Distributed Systems, Digital Subscriber Line (DSL), Python (Programming Language), Tensorflow, Software Engineering, Pytorch, Deep Learning, Scikit Learn, Information Technology
- **Published:** August 28, 2026
- **Apply:** https://nvidia.wd5.myworkdayjobs.com/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Software-Engineer--JAX_JR2004629

## About the Role

* 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.

Ways To Stand Out From The Crowd:

 * 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.



## Related Videos

- [Overview of Machine Learning in Python](https://www.wearedevelopers.com/videos/840-overview-of-machine-learning-in-python)
- [Getting Started with Machine Learning](https://www.wearedevelopers.com/videos/260-getting-started-with-machine-learning)
- [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding)
- [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs)
- [Coffee with Developers - Stephen Jones - NVIDIA](https://www.wearedevelopers.com/videos/1303-coffee-with-developers-stephen-jones-nvidia)
- [Developing an AI.SDK](https://www.wearedevelopers.com/videos/198-developing-an-ai-sdk)

## Related Articles

- [How We Built a Worry-Free System That Runs for 10+ Years – And What We’d Do Again](https://www.wearedevelopers.com/magazine/751-how-we-built-a-worry-free-system-that-runs-for-10-years-and-what-we-d-do-again)
- [What’s the latest in NVIDIA CUDA Python](https://www.wearedevelopers.com/magazine/568-what-s-the-latest-in-nvidia-cuda-python)
- [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)
- [The Best X (Twitter) Accounts for Developers](https://www.wearedevelopers.com/magazine/294-the-best-x-twitter-accounts-for-developers)
- [Best US AI Conferences for CTOs in 2026: Build vs. Buy, Vendor Evaluation, and Peer Intelligence](https://www.wearedevelopers.com/magazine/736-best-us-ai-conferences-for-ctos-in-2026-build-vs-buy-vendor-evaluation-and-peer-intelligence)
- [Résumé-Driven Development: How IT trends affect the job market for software developers](https://www.wearedevelopers.com/magazine/59-resume-driven-development-how-it-trends-affect-the-job-market-for-software-developers)