Senior Compiler Engineer — Large Scale Ai Training Perth

Senior Compiler Engineer — Large Scale Ai Training Perth

13 Nov
|
Nvidia
|
Perth

13 Nov

Nvidia

Perth

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.

📌 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.

Subscribe to this job alert:

Get the latest job offers by email for: senior compiler engineer — large scale ai training perth / perth

Subscribe to this job alert:

Get the latest job offers by email for: senior compiler engineer — large scale ai training perth / perth