Werkstudent:in (m/w/d) Softwareentwicklung für KI-Hardware (NVIDIA, Halio, Tenstorrent)

Stellenbeschreibung:

PROMED ist ein international tätiges Unternehmen im Bereich Test- und Prüfsysteme für die Elektroindustrie. Unsere Schwerpunkte sind die Entwicklung und Produktion von automatisierten Prüfsystemen, sowie der Prototypen- und Kleinserienbau. Seit über 30 Jahren bieten wir Prüfkonzepte und Lösungen für alle Bereiche der Industrie an.

Wir entwickeln hochmoderne KI-Systeme, die auf neuester Hardware laufen. Unser Fokus liegt auf der Integration leistungsfähiger KI-Hardwareplattformen wie NVIDIA Jetson, Halio-8 und Tenstorrent Wormhole/Blackhole Systeme in produktionsreife Softwaresysteme.

Zur Unterstützung unseres Entwicklerteams suchen wir einen engagierten Werkstudenten (m/w/d), der Spaß an Embedded Systems, Low-Level-Programmierung und Systemintegration hat.

Aufgaben

Entwicklung von Softwarekomponenten zur Ansteuerung und Optimierung von KI-Hardware (z.B. NVIDIA Jetson, Halio TPUs, Tenstorrent Blackhole)
Integration von Hardwarebeschleunigern in bestehende Softwaresysteme (z.B. über Schnittstellen, APIs oder Treiber)
Unterstützung bei der Entwicklung von Pipelines für Machine-Learning-Workloads (z.B. TensorFlow, PyTorch) auf Edge Devices
Durchführung von Performance-Analysen und Optimierungen auf Hardwareebene
Enge Zusammenarbeit mit unserem Softwareteam

Qualifikation

Immatrikulierter Student (m/w/d) der Informatik, Elektrotechnik, Mathematik oder vergleichbarer Studiengang
Erste Erfahrungen mit Embedded Systems oder hardwarenaher Softwareentwicklung (z.B. C/C++, CUDA, OpenCL)
Kenntnisse in Python
Interesse an KI-Hardware und Machine-Learning-Frameworks (z.B. TensorFlow Lite, ONNX, PyTorch)
Kenntnisse in Linux und Systemprogrammierung wünschenswert
Selbstständige und strukturierte Arbeitsweise