> Markdown version of [/jobs/ext/3721528-senior-software-engineer-jax](https://www.wearedevelopers.com/jobs/ext/3721528-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
- **Location:** UK
- **Experience:** Expert
- **Salary:** £292,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:** October 14, 2026
- **Apply:** https://www.totaljobs.com/job/senior-software-engineer/nvidia-job108161939

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