Senior Compiler Engineer — Large-Scale AI Training (Australia)

Senior Compiler Engineer — Large-Scale AI Training (Australia)

12 Nov
|
Nvidia
|
Australia

12 Nov

Nvidia

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 JaxPP into training frameworks, and contribute to open-source initiatives.

#J-18808-Ljbffr

📌 Senior Compiler Engineer — Large-Scale AI Training (Australia)
🏢 Nvidia
📍 Australia

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.

Subscribe to this job alert:

Get the latest job offers by email for: senior compiler engineer — large-scale ai training (australia) / australia

Subscribe to this job alert:

Get the latest job offers by email for: senior compiler engineer — large-scale ai training (australia) / australia