Browse tech roles

Filter the feed by workplace, employment type, salary floor, and post age. For ranked matching against your resume, use AI Match.

Active Keyword: JAX Remove filter Clear all

3 of up to 20 (filtered)

Senior Software Engineer, JAX

Nvidia

Remote (Santa Clara, CA) 55 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 59 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) 143 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