Arbeitsort: Remote (Schweiz)Pensum: 100%Aufgaben:Sie entwickeln die KI-Plattform eines weltweit führenden Technologieunternehmens weiter, insbesondere Performance-Optimierungen in Deep-Learning-Frameworks auf Basis von JAX. Sie tragen massgeblich zur Weiterentwicklung von JAX bei, entwerfen und implementieren JAX-Kernkomponenten und holen das Maximum an Performance aus GPU-Produkten heraus. Sie arbeiten mit angewandten KI-Forschenden an zukunftssicheren Modellen, bauen Tools zur Effizienzsteigerung von Teams, die KI-Systeme entwickeln, und schlagen die Brücke zwischen aktueller Forschung in Numerik, Simulation und Deep Learning und deren Anwendung in realen Produkten.Profil:Sie haben einen Bachelor in Informatik oder Computer Engineering (oder gleichwertige Erfahrung) und mindestens 6 Jahre relevante Erfahrung. Sie programmieren in C/C++ und Python, kennen Machine-Learning-Frameworks und deren Interna (z. B. PyTorch, TensorFlow, scikit-learn), haben kundenorientierte Lösungen entwickelt und verfügen über ein fundiertes Verständnis von CPU- und GPU-Architekturen, numerischen Bibliotheken und modularem Softwaredesign. Von Vorteil sind Kenntnisse von JAX, Autograd, Tracing, Codegenerierung und DSL-Compilern, verteiltem Deep-Learning-Training (Multi-GPU, Multi-Node), Software-Release-Zyklen sowie grossskaligen verteilten Systemen. Sie kommunizieren ausgezeichnet auf Englisch und arbeiten gerne in internationalen, funktionsübergreifenden Teams.Geboten werden die Möglichkeit, die nächste Generation von Tools für numerisches Rechnen und Machine-Learning-Forschung weltweit mitzugestalten, vollständig remote zu arbeiten sowie ein international vernetztes Umfeld mit marktgerechter Entlöhnung.
Benötigte Skills
- Englisch
Pensum Vollzeit
#J-18808-Ljbffr
Senior Software Engineer JAX / Deep Learning Frameworks 100% in Zürich Arbeitgeber: myitjob GmbH
Als Embedded Linux Developer (m/w) in Zürich bieten wir Ihnen die Möglichkeit, in einem dynamischen und innovativen Umfeld zu arbeiten, das auf Teamarbeit und interdisziplinäre Zusammenarbeit setzt. Unsere Unternehmenskultur fördert kontinuierliches Lernen und persönliche Entwicklung, während wir Ihnen attraktive Benefits und eine ausgewogene Work-Life-Balance bieten. Nutzen Sie die Chance, an spannenden Projekten im Bereich IoT und industrielle Kommunikation mitzuwirken und Ihre Fähigkeiten in einem zukunftsorientierten Unternehmen weiter auszubauen.
