Neuer Job?Nejo!

Die KI-Suchmaschine für Jobs

NV
NVIDIA
vor 3 Monaten

Senior Deep Learning Compiler Engineer - PyTorch(m/w/x)

Berlin
VollzeitVor OrtSenior
AI/ML

Beschreibung

In this role, you will be at the cutting edge of AI compiler technology, driving the design and optimization of tools that enhance deep learning performance. Your work will involve collaboration with top engineers and performance analysis to shape the future of the PyTorch ecosystem.

Lass KI die perfekten Jobs für dich finden!

Lade deinen CV hoch und die Nejo-KI findet passende Stellenangebote für dich.

Anforderungen

  • Bachelor's, Master's, or Ph.D. in Computer Science or related field, or equivalent experience
  • 8+ years of relevant work experience
  • Strong command of Python
  • Experience with deep learning frameworks like PyTorch or JAX
  • Solid foundation in compiler concepts
  • Excellent communication and collaboration skills
  • Previous contributions to deep learning compiler projects or frameworks
  • Deep expertise in PyTorch internals
  • Experience with JAX-like functional transformations
  • Familiarity with parallel programming and distributed systems
  • Track record of impactful participation in open-source communities

Ausbildung

Bachelor-Abschluss
ODER
Master-Abschluss
ODER
Doktor / Ph.D.

Berufserfahrung

8 Jahre

Aufgaben

  • Lead the design and implementation of core compiler technologies
  • Optimize and maintain compiler technologies for deep learning workloads
  • Conduct performance analysis on workloads running on thousands of GPUs
  • Identify optimization opportunities to enhance Thunder's design
  • Collaborate with engineers who developed PyTorch for NVIDIA hardware
  • Pioneer new features in compiler innovation
  • Work with compiler, library, and systems teams to translate research into solutions

Tools & Technologien

PythonPyTorchJAXCUDA

Sprachen

Englischverhandlungssicher

Die Originalanzeige dieses Stellenangebotes in der aktuellsten Version findest du hier. Nejo hat diesen Job automatisch von der Website des Unternehmens NVIDIA erfasst und die Informationen auf Nejo mit Hilfe von KI für dich aufbereitet. Trotz sorgfältiger Analyse können einzelne Informationen unvollständig oder ungenau sein. Bitte prüfe immer alle Angaben in der Originalanzeige! Inhalte und Urheberrechte der Originalanzeige liegen beim ausschreibenden Unternehmen.
Noch nicht perfekt?
100+ Ähnliche Jobs in Berlin
  • NVIDIA

    Senior Developer Technology Engineer, Artificial Intelligence(m/w/x)

    Vollzeitnur vor OrtSenior
    Berlin
  • NVIDIA

    Principal AI Developer Technology Engineer(m/w/x)

    Vollzeitnur vor OrtSenior
    Berlin
  • Perplexity

    Machine Learning Research Engineer - Search(m/w/x)

    Vollzeitnur vor OrtBerufserfahren
    Berlin
  • AUTO1 Group

    Senior Computer Vision Engineer(m/w/x)

    Vollzeitnur vor OrtSenior
    Berlin
  • RepRisk AG

    Senior AI Engineer(m/w/x)

    Vollzeitnur vor OrtSenior
    Berlin
100+ Alle ähnlichen Jobs ansehen