Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance

Dołącz do naszego zespołu kompilatora PyTorch w Meta

Przegląd:

Zespół kompilatora PyTorch w Meta ma misję polegającą na poprawie wydajności PyTorch, koncentrując się na efektywności i prędkości, bez naruszania jego renomowanej elastyczności. Nasza przełomowa technologia PT2 od momentu swojego powstania w marcu 2023 roku ustawiła nowy punkt odniesienia w krajobrazie frameworków ML, popychając PyTorch w kierunku stania się wiodącym kompilatorem ML w branży. Poszukujemy Inżyniera Oprogramowania, który przyczyni się do rozwoju kompilatora PT2 i przesunie granice technologii uczenia maszynowego.

Kluczowe Obowiązki:

• Rozwijanie i ulepszanie komponentów kompilatora PT2, takich jak TorchDynamo, TorchInductor i PyTorch Core.
• Stymulowanie ulepszeń wydajności PyTorch, dostarczając systematyczne rozwiązania, które przynoszą korzyści globalnej społeczności PyTorch.
• Badania i optymalizacja modeli AI generatywnych, skupiając się na etapach pre-treningu, fine-tuningu i inferencji.
• Poszukiwanie synergii między kompilatorem PyTorch a rozproszonymi systemami PyTorch.
• Aktywna współpraca z użytkownikami PyTorch w celu wspierania adopcji i innowacyjnego wykorzystania technologii PT2.

Minimalne Kwalifikacje:

• BS/MS w dziedzinie Informatyki lub pokrewnej.
• Doświadczenie praktyczne w kompilatorach, systemach ML, obliczeniach wysokiej wydajności i optymalizacji GPU.
• Znajomość programowania w Pythonie lub CUDA.
• Studiowanie lub uzyskanie stopnia licencjata z informatyki lub pokrewnej dziedziny technicznej.

Preferowane Kwalifikacje:

• Doktorat z Informatyki, Matematyki Stosowanej lub pokrewnej dziedziny.
• Doświadczenie z technologiami PT2, Triton, MLIR lub bezpośrednie wkłady do PyTorch.
• Głęboka specjalizacja w wydajności GPU i rozwoju wysokowydajnych jąder CUDA.
• Udowodnione osiągnięcia w badaniach i inżynierii oprogramowania, potwierdzone publikacjami, patentami lub udziałem w odpowiednich konkursach.

O Meta:

Meta nadal przedefiniowuje krajobraz łączności i interakcji społecznych za pomocą innowacyjnych technologii. Od uruchomienia Facebooka w 2004 roku i kolejnych aplikacji takich jak Instagram i WhatsApp, teraz przenosimy się do rzeczywistości rozszerzonej i wirtualnej, aby inżynierować nową erę technologii społecznościowej. W Meta, twoja praca przekracza tradycyjne cyfrowe granice, tworząc nowe sposoby łączenia się ludzi, rozwijania społeczności i wzrostu firm.

Wynagrodzenie i Korzyści:

Rola oferuje konkurencyjne wynagrodzenie w przedziale 117 000 do 173 000 dolarów rocznie, plus bonus