
Senior Software Engineer, JAX
NVIDIA is hiring a Senior Software Engineer to design and implement JAX core components and drive performance on NVIDIA hardware. Responsibilities include collaborating with applied researchers, building developer tools, and bridging numerical computing and deep learning research to products. Requirements include a BS or equivalent, 6+ years relevant experience, strong C/C++ and Python programming, knowledge of ML frameworks (e.g., PyTorch, TensorFlow), familiarity with CPU/GPU architectures and numeric libraries, and strong system/software design and communication skills. Poland-specific base salary ranges are provided in the posting.











