TPU ML Performance Engineer — JAX & Kernel Optimizer

TPU ML Performance Engineer — JAX & Kernel Optimizer

Zürich Vollzeit 45000 - 65000 € / Jahr (geschätzt) Kein Homeoffice möglich
P

Auf einen Blick

  • Aufgaben: Optimiere JAX-Workloads auf TPUs und entwickle Kernel für maximale Effizienz.
  • Unternehmen: Pageshift, ein innovatives Forschungszentrum in Zürich.
  • Vorteile: Wettbewerbsfähiges Gehalt, flexible Arbeitszeiten und spannende Projekte.
  • Weitere Informationen: Dynamisches Team mit Fokus auf Forschung und Entwicklung.
  • Warum dieser Job: Gestalte die Zukunft der KI mit modernster Technologie und kreativen Lösungen.
  • Qualifikationen: Grundkenntnisse in TPUs, JAX und XLA; Erfahrung mit Pallas-Kernels von Vorteil.

Das prognostizierte Gehalt liegt zwischen 45000 - 65000 € pro Jahr.

Pageshift, ein Forschungszentrum in Zürich, sucht eine Person zur Optimierung von JAX-Workloads auf TPUs mit Schwerpunkt auf der Entwicklung von Kernen und der Effizienz von End-to-End-Prozessen. Der erfolgreiche Kandidat wird verantwortlich sein für:

  • Implementierung von Optimierungen auf Kernel-Ebene
  • Profilierung und Validierung von Verbesserungen mit Daten

Bewerber sollten eine Leidenschaft für Storytelling sowie ein grundlegendes Verständnis von TPUs, JAX und XLA mitbringen. Erfahrung im Schreiben von Pallas-Kernen ist von Vorteil und verbessert Ihre Bewerbung.

TPU ML Performance Engineer — JAX & Kernel Optimizer Arbeitgeber: Pageshift

Pageshift ist ein hervorragender Arbeitgeber, der eine inspirierende Arbeitsumgebung in Zürich bietet, wo Innovation und Kreativität gefördert werden. Unsere Mitarbeiter profitieren von flexiblen Arbeitszeiten, einer offenen Unternehmenskultur und umfangreichen Möglichkeiten zur beruflichen Weiterentwicklung, während sie an spannenden Projekten im Bereich der TPU-Optimierung arbeiten. Bei uns haben Sie die Chance, Ihre Fähigkeiten in einem dynamischen Team einzubringen und an der Spitze der Technologie zu stehen.

P

Kontaktdaten:

Pageshift Recruiting-Team

StudySmarter Expertenrat🤫

Wir sind der Meinung, dass Sie so TPU ML Performance Engineer — JAX & Kernel Optimizer erhalten könnten

Tipp Nummer 1

Mach dir eine Liste von Unternehmen, die an deiner Expertise interessiert sein könnten. Wenn du dich auf die Stellen bewirbst, die wirklich zu deinen Fähigkeiten passen, steigert das deine Chancen, eingeladen zu werden.

Tipp Nummer 2

Nutze dein Netzwerk! Sprich mit Leuten, die in der Branche arbeiten oder bereits bei Pageshift sind. Oft gibt es interne Empfehlungen, die dir einen Vorteil verschaffen können.

Tipp Nummer 3

Bereite dich gut auf Vorstellungsgespräche vor. Informiere dich über die neuesten Entwicklungen in JAX und TPUs. Zeige, dass du nicht nur die technischen Fähigkeiten hast, sondern auch leidenschaftlich über das Thema bist.

Tipp Nummer 4

Bewirb dich direkt über unsere Website! Das zeigt dein Interesse und du kannst sicherstellen, dass deine Bewerbung die richtige Anlaufstelle erreicht. Lass uns gemeinsam den nächsten Schritt in deiner Karriere gehen!

Wir glauben, dass du diese Fähigkeiten brauchst, um TPU ML Performance Engineer — JAX & Kernel Optimizer mit Bravour zu bestehen

Kernel-Entwicklung
JAX
TPU-Optimierung
XLA
Profiling
Datenvalidierung
Pallas-Kernel-Schreiben

Einige Tipps für deine Bewerbung 🫡

Sei authentisch!:Zeig uns, wer du wirklich bist! Deine Persönlichkeit und Leidenschaft für das Thema sind genauso wichtig wie deine technischen Fähigkeiten. Lass uns in deinem Anschreiben spüren, warum du dich für die Position als TPU ML Performance Engineer interessierst.

Betone deine Erfahrungen:Wenn du schon mit JAX, TPUs oder Pallas-Kernen gearbeitet hast, dann erzähl uns davon! Konkrete Beispiele aus deiner bisherigen Arbeit helfen uns, deine Eignung besser zu verstehen und zeigen, dass du die nötigen Skills mitbringst.

Mach es strukturiert:Halte deine Bewerbung klar und übersichtlich. Verwende Absätze und Aufzählungen, um wichtige Informationen hervorzuheben. So können wir schnell erkennen, was du drauf hast und was dich auszeichnet.

Bewirb dich über unsere Website:Wir empfehlen dir, deine Bewerbung direkt über unsere Website einzureichen. Das macht es uns einfacher, alles zu verwalten und sicherzustellen, dass deine Unterlagen nicht verloren gehen. Also, leg los und bewirb dich!

Wie man sich auf ein Vorstellungsgespräch bei Pageshift vorbereitet

Verstehe die Technologie

Mach dich mit TPUs, JAX und XLA vertraut. Lies aktuelle Artikel oder schaue Tutorials, um ein tiefes Verständnis für die Technologien zu entwickeln, die du optimieren wirst. Das zeigt dein Engagement und deine Leidenschaft für das Thema.

Bereite konkrete Beispiele vor

Überlege dir spezifische Projekte oder Erfahrungen, in denen du Kernel-Optimierungen durchgeführt hast. Sei bereit, diese Beispiele im Interview zu erläutern und zu zeigen, wie du Probleme gelöst hast. Das macht deinen Hintergrund greifbarer.

Storytelling ist der Schlüssel

Da eine Leidenschaft für Storytelling gefordert wird, übe, deine technischen Erfahrungen in eine spannende Geschichte zu verpacken. Erkläre, wie du Herausforderungen gemeistert hast und welche Ergebnisse du erzielt hast. So bleibst du im Gedächtnis.

Fragen stellen

Bereite einige Fragen vor, die du dem Interviewer stellen kannst. Das zeigt dein Interesse an der Position und hilft dir, mehr über die Unternehmenskultur und die Erwartungen zu erfahren. Frag nach den aktuellen Projekten oder Herausforderungen im Team.