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 JaxPP into training frameworks, and contribute to open-source initiatives.
#J-18808-Ljbffr
📌 Senior Compiler Engineer — Large-Scale AI Training (Perth)
🏢 Nvidia
📍 Perth
Reply to this offer
Impress this employer describing Your skills and abilities, fill out the form below and leave Your personal touch in the presentation letter.