Senior Deep Learning Compiler Ingegnere - PyTorch — NVIDIA (ufficio Zurich)

CHF 101'500 - 154'000
NVIDIA (ufficio Zurich) · Zürich (ZH)
Categoria: Ingegneria Contratto: full-time Salario: CHF 101'500 - 154'000
Vai alla candidatura
Località
Zürich
Contratto
full-time
Pubblicato
31 giorni fa
SalarioCHF 101'500 - 154'000

Panoramica

Unisciti a noi all'avanguardia della tecnologia AI compilatore e contribuire a modellare il futuro del calcolo accelerato.

NVIDIA è alla ricerca di ingegneri appassionati per costruire la prossima generazione di strumenti utilizzati da sviluppatori AI e ricercatori in tutto il mondo.

Il nostro team sta sviluppando Thunder , un ambizioso compilatore sorgente-fonte costruito per sbloccare prestazioni eccezionali per i modelli PyTorch su GPU NVIDIA.

Requisiti principali

  • Ecco alcune delle competenze che vorremmo vedere:
  • Corso di laurea, Master, o Ph.D. in Informatica o un campo tecnico correlato (o esperienza equivalente).
  • 8+ anni di esperienza lavorativa rilevante
  • Un forte comando di Python ed esperienza costruzione di sistemi software complessi e ben testati.
  • Capisci come i modelli sono costruiti e dove le sfide di prestazione si trovano.
  • Una solida base in concetti di compilatore come alberi di sintassi astratti (AST), rappresentazioni intermedie (ad esempio, forma SSA), analisi di programma e generazione di codice.
  • Eccellente capacità di comunicazione e collaborazione, essenziale per lavorare efficacemente in un ambiente distribuito e open source. Modi per distinguersi dalla folla:
  • Contributi precedenti a progetti di compilatore di apprendimento profondo (ad esempio, TVM, MLIR, IREE) o framework di apprendimento profondo stessi.
  • Profonda esperienza negli interni di PyTorch, in particolare il suo stack di compilatori (TorchDynamo, TorchInductor).
  • Esperienza con trasformazioni funzionali simili a JAX e la loro applicazione in un contesto compilatore.
  • Familiarità con programmazione parallela, sistemi distribuiti e scrittura codice CUDA ad alte prestazioni.
  • Un track record di partecipazione impattante alle comunità open source, come ad esempio attraverso contributi di codice, discussioni di progettazione o mentorship.

Cosa offre l’azienda

  • Ti immergerai nell'analisi delle prestazioni, scrutando i carichi di lavoro in esecuzione su migliaia di GPU per trovare opportunità di ottimizzazione che modellano il futuro design di Thunder.
  • Inoltre, sarai parte di un ecosistema vibrante, lavorando a stretto contatto con i principali team di compilatori, biblioteche e sistemi, tra cui esperti dietro nvFuser, TVM, XLA e CUDA, per tradurre le ultime ricerche in soluzioni pratiche e ad alto impatto per la comunità open source. Cosa dobbiamo vedere:
  • Siamo alla ricerca di ingegneri che sono entusiasti di costruire strumenti potenti e incentrati sull'utente e sono comodi a lavorare in un ambiente veloce e collaborativo.
  • stipendi altamente competitivi, un pacchetto di benefici e un ambiente di lavoro che promuove la diversità, l'inclusione e la flessibilità.
  • Come datore di lavoro pari opportunità, ci impegniamo a promuovere un posto di lavoro di supporto e di potenziamento per tutti.

Azienda e contesto

  • Esperienza pratica con strutture di apprendimento profonde come PyTorch o JAX.

Dettagli ulteriori

  • Se sei spinto a risolvere problemi complessi e vuoi fare un impatto fondamentale sull'ecosistema AI, applica per unire il nostro team collaborativo e innovativo.
  • Inoltre, sarai parte di un ecosistema vibrante, lavorando a stretto contatto con i principali team di compilatori, biblioteche e sistemi, tra cui esperti dietro nvFuser, TVM, XLA e CUDA, per tradurre le ultime ricerche in soluzioni pratiche e ad alto impatto per la comunità open source. Cosa dobbiamo vedere:
  • Eccellente capacità di comunicazione e collaborazione, essenziale per lavorare efficacemente in un ambiente distribuito e open source. Modi per distinguersi dalla folla:
  • Offriamo stipendi altamente competitivi, un pacchetto di benefici e un ambiente di lavoro che promuove la diversità, l'inclusione e la flessibilità.
  • Il tuo stipendio base sarà determinato in base alla tua posizione, esperienza e la paga dei dipendenti in posizioni simili. Per la Polonia: la base sal

Note e contenuto originale

  • Cosa farai:
  • Inoltre, sarai parte di un ecosistema vibrante, lavorando a stretto contatto con i principali team di compilatori, biblioteche e sistemi, tra cui esperti dietro nvFuser, TVM, XLA e CUDA, per tradurre le ultime ricerche in soluzioni pratiche e ad alto impatto per la comunità open source.
  • Cosa dobbiamo vedere:
  • Eccellente capacità di comunicazione e collaborazione, essenziale per lavorare efficacemente in un ambiente distribuito e open source.
  • Modi per distinguersi dalla folla:
  • Il tuo stipendio base sarà determinato in base alla tua posizione, esperienza e la paga dei dipendenti in posizioni simili.
  • Per la Polonia: la base sal
Vai alla candidatura
Logo NVIDIA (ufficio Zurich)
Azienda
NVIDIA (ufficio Zurich) · Zürich
Frontaliere Ticino ha scovato questa opportunità nel monitoraggio aziende.

Tutte le offerte NVIDIA (ufficio Zurich) Zürich →

Esplora annunci simili