Senior Software Engineer JAX / Deep Learning Frameworks 100%

Stellenbeschreibung

###### Job Informationen ###### 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 * Python * Machine Learning * C * C++ * Bachelor * Master