W skrócie
- Zadania: Twórz innowacyjne narzędzia kompilacyjne dla modeli PyTorch na GPU NVIDIA.
- Firma: NVIDIA - lider w technologii AI i obliczeniach przyspieszonych.
- Benefity: Konkurencyjne wynagrodzenie, bogaty pakiet benefitów i elastyczne warunki pracy.
- Pozostałe informacje: Możliwość rozwoju kariery w różnorodnym i wspierającym środowisku.
- Dlaczego ta praca: Pracuj nad przełomowymi projektami w dynamicznym zespole ekspertów.
- Kwalifikacje: Doświadczenie w programowaniu w Pythonie i frameworkach deep learning.
Przewidywane wynagrodzenie mieści się w przedziale 135000 - 165000 zł za rok.
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, aplikuj, aby dołączyć do naszego współpracującego i innowacyjnego zespołu.
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. Będziesz głęboko analizować 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. Ponadto będziesz częścią żywego 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:
Poszukujemy inżynierów, którzy są podekscytowani budowaniem potężnych, skoncentrowanych na użytkownikach narzędzi i czują się komfortowo w szybkim, współpracującym środowisku. Oto niektóre z umiejętności, które chcielibyśmy zobaczyć:
- Licencjat, magister lub doktorat w dziedzinie informatyki lub pokrewnej dziedziny technicznej (lub równoważne doświadczenie).
- Ponad 8-letnie doświadczenie zawodowe.
- 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 generowanie 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.
Senior Deep Learning Compiler Engineer in Katowice – pracodawca: NVIDIA Corporation
NVIDIA to doskonały pracodawca, oferujący wyjątkowe możliwości rozwoju zawodowego w dynamicznej i innowacyjnej kulturze pracy. Pracując w zespole inżynierów zajmujących się bibliotekami matematycznymi, będziesz miał okazję wpływać na rozwój technologii AI i HPC, a także korzystać z konkurencyjnych wynagrodzeń oraz bogatego pakietu benefitów. Nasza lokalizacja w Polsce sprzyja elastycznemu modelowi pracy, co pozwala na doskonałą równowagę między życiem zawodowym a prywatnym.
