Senior Deep Learning Compiler Engineer - PyTorch

Polska
24 000 – 42 000 PLN
netto / mies.
UoPRemote
PyTorchPythonCUDATVMMLIRJAXTorchDynamoTorchInductorIREEXLA

Dołączysz do zespołu NVIDIA rozwijającego Thunder - kompilator źródło-do-źródła dla modeli PyTorch na GPU NVIDIA.

Kluczowe obszary

Projektowanie i implementacja technologii kompilatorowych (optymalizacja i utrzymanie)
Analiza wydajności workloadów AI (optymalizacja dla tysięcy GPU)
Współpraca z zespołami PyTorch (rozwój nowych funkcji dla hardware NVIDIA)
Praca z ekosystemem kompilatorów (nvFuser, TVM, XLA, CUDA)
Opis na podstawie oferty pracodawcy · Nvidia