> Markdown version of [/jobs/ext/3736942-software-engineer](https://www.wearedevelopers.com/jobs/ext/3736942-software-engineer). 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).

---

# Software Engineer

- **Company:** NVIDIA Corporation
- **Location:** Jacksonville, FL, United States
- **Experience:** Expert
- **Salary:** $184,000.0 - $287,500.0
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, C++ (Programming Language), Compilers, Code Generation, Computer Engineering, Distributed Systems, Digital Subscriber Line (DSL), Python (Programming Language), Tensorflow, Software Engineering, Pytorch, Deep Learning, Numerical Computing, Scikit Learn, Information Technology, Build Tools
- **Published:** October 14, 2026
- **Apply:** https://www.juju.com/job/1_ScrapingHub33470-JR2004629

## About the Role

What We Need To See: + 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.

## Description

What You Will Be Doing: + Play meaningful role in NVIDIA's effort in contributing to JAX. + Design and implement JAX core components and drive peak performance on NVIDIA products. + Work with AI applied researchers and leaders to build future-proof models + Build tools that will increase the efficiency of teams developing AI-based systems. + Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products.