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

2 of up to 20 (filtered)

Senior Software Engineer, JAX

Nvidia

Remote (Santa Clara, CA) 16 days ago $184,000–$287,500
Actively hiring Confirmed live today High trust Above market
JAX Python C/C++ PyTorch TensorFlow scikit-learn NumPy GPU Architecture Distributed Systems Numerical Computing DSL Compilers Autograd Machine Learning Software Design
6+ yrs exp Remote
Save

Senior Manager, Software Engineering

Nvidia

Santa Clara, CA 85 days ago $272,000–$431,250
Actively hiring Confirmed live today High trust Above market
JAX XLA CUDA HLO PJRT Shardy MPMD TileIR PyTorch TensorFlow cuDNN NCCL TensorRT Transformer Engine High-Performance Computing Inference Robotics GPU Programming System Software
10+ yrs exp
Save