Software Engineer, PyTorch - Deep Learning

Firma: NVIDIA

Stanowisko: Inżynier Oprogramowania, PyTorch - Głębokie Uczenie

Lokalizacja: Otwarte dla Globalnych Talentów

Jesteś gotowy do innowacji na przecięciu AI i inżynierii oprogramowania? NVIDIA poszukuje wykwalifikowanych Inżynierów Kompilatorów Głębokiego Uczenia do tworzenia najnowocześniejszych narzędzi, które zwiększają możliwości inżynierów AI na całym świecie. Zgarnij okazję, aby wpłynąć na projektowanie, rozwój i wdrażanie AI na skalę globalną, wykorzystując tysiące GPU. Dołącz do naszego zespołu, który koncentruje się na dostarczaniu nieustającej doskonałości w doświadczeniach z oprogramowaniem na wiodącym sprzęcie NVIDIA z PyTorch.

Jako kluczowy członek naszego zespołu, twoja rola będzie sięgać poza tradycyjne granice, napędzając ewolucję rozproszonych obciążeń treningowych głębokiego uczenia za pomocą zaawansowanych technologii kompilatorów. Twoje podstawowe obowiązki będą obejmowały:

• Wykonywanie dogłębnej analizy obciążeń pracy głębokiego uczenia, aby zlokalizować i rozwiązać problemy z wydajnością, anomalią funkcjonalną i nieefektywnością systemu.
• Łączenie problemów z wydajnością bezpośrednio z defektami kompilatora lub nieodkrytym potencjałem optymalizacji i formułowanie solidnych rozwiązań.
• Praca wspólnie nad doskonaleniem transformacji programów lub innowacją nowych, na podstawie kompleksowych przeglądów wydajności.
• Bycie na bieżąco z najnowszymi rozwojami w kompilatorach głębokiego uczenia, aby proponować najnowocześniejsze rozwiązania, które zwiększają efektywność frameworków głębokiego uczenia.
• Przeprowadzenie dokładnych testów i walidacji, aby zapewnić optymalną wydajność i jakość w treningu modeli.

Zapraszamy z:

• Bakalaureatem, magistrem lub doktorem z informatyki lub w podobnym technicznym kierunku, lub porównywalnym doświadczeniem.
• Znajomością Pythona i znajomością frameworków uczenia maszynowego, takich jak PyTorch lub JAX.
• Solidnym zrozumieniem koncepcji kompilatorów, w tym abstrakcyjnej interpretacji, reprezentacji kodu takich jak forma SSA i AST, generowania kodu i transformacji programu.
• Udokumentowanym doświadczeniem w rozwijaniu dużych projektów oprogramowania.
• Mocnymi umiejętnościami komunikacyjnymi zarówno w mowie, jak i piśmie.

Wyróżnij się z:

• Wcześniejszymi wkładami do projektów kompilatora głębokiego uczenia typu open-source, takich jak TVM, lub udziałem w frameworkach głębokiego uczenia.
• Zaawansowaną znajomością wewnętrznych mechanizmów PyTorch i/lub JAX. <