Browse tech roles

Basic role filtering by workplace, salary floor, and post age. For full AI matching and advanced filtering upload your resume using AI Match.

Active Keyword: jax Remove filter Clear all

4 of up to 20 (filtered)

Senior Software Engineer, JAX

Nvidia

Remote (Santa Clara, CA) 64 days ago $184,000–$287,500
Actively hiring Above market
JAX Python C++ PyTorch TensorFlow scikit-learn CI/CD GPU CPU Docker Kubernetes AWS PostgreSQL Prometheus Grafana
Remote
Save

Senior Manager, Software Engineering - JAX

Nvidia

Santa Clara, CA 68 days ago $272,000–$431,250
Actively hiring Above market
JAX XLA CUDA TensorRT PyTorch TensorFlow Python C++ GPU CPU LPU CI/CD NVIDIA heterogeneous computing high-performance computing deep learning frameworks distributed systems performance optimization software engineering methodologies
Save

Distinguished Engineer, JAX

Nvidia

Remote (Santa Clara, CA) 152 days ago $320,000–$488,750
Actively hiring Above market
JAX Python C/C++ PyTorch TensorFlow scikit-learn GPU CPU Distributed Systems CI/CD Autograd DSL Compilers Multi-GPU Multi-node Synchronous Communication Asynchronous Communication
Remote
Save