Senior Deep Learning Compiler Engineer - PyTorch in Warsaw

Dołącz do nas na czołowej pozycji w technologii kompilatorów AI i pomóż kształtować przyszłość przyspieszonego obliczania. NVIDIA poszukuje pasjonujących inżynierów do budowy następnej generacji narzędzi używanych przez deweloperów i badaczy AI na całym świecie. Nasz zespół rozwija Thunder, ambitny kompilator typu source-to-source, który ma na celu odblokowanie wyjątkowej wydajności modeli PyTorch na procesorach graficznych NVIDIA.

To unikalna okazja, aby przyczynić się do projektu, który wzbogaca ekosystem PyTorch, pracując z nowoczesnymi stosami kompilatorów, takimi jak TorchDynamo i TorchInductor, aby tworzyć potężne, otwarte rozwiązania, które przynoszą korzyści całej społeczności. Jeśli jesteś zmotywowany do rozwiązywania złożonych problemów i chcesz mieć fundamentalny wpływ na ekosystem AI, to jest to rola dla Ciebie.

Co będziesz robić:

• Jako kluczowy członek naszego zespołu, będziesz bezpośrednio przyczyniać się do przyszłości przyspieszonej AI.
• Twoja rola będzie dynamiczna i głęboko techniczna, umieszczając Cię w centrum innowacji kompilatorów.
• Będziesz prowadzić projektowanie, implementację, optymalizację i utrzymanie podstawowych technologii kompilatorów, które przyspieszają ogromne obciążenia związane z uczeniem głębokim.
• To bardzo współpraca rola, w której będziesz pracować obok inżynierów, którzy stworzyli PyTorch dla sprzętu NVIDIA, pomagając w pionierskich nowych funkcjach i pozostając na czołowej pozycji w rozwoju frameworków.
• Przeanalizujesz wydajność, badając obciążenia działające na tysiącach GPU, aby znaleźć możliwości optymalizacji, które ukształtują przyszły projekt Thunder.
• Będziesz częścią dynamicznego ekosystemu, ściśle współpracując z wiodącymi zespołami kompilatorów, bibliotek i systemów, w tym ekspertami stojącymi za nvFuser, TVM, XLA i CUDA, aby przetłumaczyć najnowsze badania na praktyczne, mające duży wpływ rozwiązania dla społeczności open-source.

Czego potrzebujemy:

• Dyplom licencjata, magistra lub doktorat w dziedzinie informatyki lub pokrewnej (lub równoważne doświadczenie).
• 8+ lat odpowiedniego doświadczenia zawodowego.
• Silna znajomość Pythona i doświadczenie w budowaniu złożonych, dobrze przetestowanych systemów oprogramowania.
• Praktyczne doświadczenie z frameworkami uczenia głębokiego, takimi jak PyTorch lub JAX.
• Rozumiesz, jak budowane są modele i gdzie leżą wyzwania wydajnościowe.
• Solidne podstawy w koncepcjach kompilatorów, takich jak drzewa składniowe (AST), reprezentacje pośrednie (np. forma SSA), analiza programów i generacja kodu.
• Doskonałe umiejętności komunikacyjne i współpracy, niezbędne do efektywnej pracy w rozproszonym, otwartym środowisku.

Sposoby, aby wyróżnić się z tłumu:

• Poprzednie wkłady w projekty kompilatorów uczenia głębokiego (np. TVM, MLIR, IREE) lub same frameworki uczenia głębokiego.
• Głęboka wiedza na temat wnętrzności PyTorch, szczególnie jego stosu kompilatorów (TorchDynamo, TorchInductor).
• Doświadczenie z transformacjami funkcyjnymi podobnymi do JAX i ich zastosowaniem w kontekście kompilatora.
• Znajomość programowania równoległego, systemów rozproszonych i pisania wysokowydajnego kodu CUDA.
• Historia znaczącego uczestnictwa w społecznościach open-source, np. poprzez wkład w kod, dyskusje projektowe lub mentoring.

NVIDIA jest na czołowej pozycji w przełomach w sztucznej inteligencji, obliczeniach o wysokiej wydajności i wizualizacji. Nasze zespoły składają się z zmotywowanych, innowacyjnych profesjonalistów, którzy dążą do przesuwania granic technologii. Oferujemy bardzo konkurencyjne wynagrodzenia, rozbudowany pakiet świadczeń oraz środowisko pracy, które promuje różnorodność, inkluzję i elastyczność. Jako pracodawca oferujący równe szanse, zobowiązujemy się do wspierania i wzmacniania miejsca pracy dla wszystkich. Twoje wynagrodzenie podstawowe zostanie ustalone na podstawie lokalizacji, doświadczenia i wynagrodzenia pracowników na podobnych stanowiskach. Dla Polski: Zakres wynagrodzenia podstawowego wynosi 292 500 PLN - 507 000 PLN.