13 Nov
|
Nvidia
|
Western Australia
13 Nov
Nvidia
Western Australia
NVIDIA Corporation is seeking a senior engineer to design and implement program transformations on JAX programs (jaxpr) for large-scale training, atop unmodified JAX and XLA.
You will help optimize compilers, build distributed runtimes, and facilitate data movement across GPU stages using NCCL.
You will collaborate with JAX/XLA teams, work on integrating Jax PP into training frameworks, and contribute to open-source initiatives.
J-*-Ljbffr
📌 Senior Compiler Engineer — Large Scale Ai Training Western Australia
🏢 Nvidia
📍 Western Australia