Software Engineering Internship – Machine Learning on GPUs

We are looking for passionate and motivated students or recent graduates in Computer Science or related fields to join our team and contribute to cutting-edge open-source machine learning (ML) projects.

This paid internship, lasting 3–6 months, with a flexible start date, offers a unique opportunity to work on real-world challenges in high-performance computing and machine learning, focusing on improving support for GPU hardware. Positions are available at our development centers in Serbia and Bosnia.

You will collaborate with experienced engineers and the open-source community to enhance GPU compatibility, improve performance, and ensure robust functionality across diverse workloads.

This role is ideal for final-year students or recent graduates eager to gain hands-on experience in systems programming, GPU computing, and machine learning frameworks, while making meaningful contributions to widely used open-source repositories.

• Systems Programming: Strong knowledge of C, C++, or other systems programming Languages
• Scripting: Experience with Python for scripting and automation
• Neural Networks: Familiarity with neural network architectures (Transformers, diffusion models, CNNs)
• Soft Skills: Strong sense of ownership, proactivity, and clear communication

• ML Frameworks: Experience with machine learning frameworks such as PyTorch, TensorFlow, ONNX Runtime, JAX, or similar
• GPU Programming: Familiarity with GPU programming (CUDA, HIP, OpenCL, SYCL) and/or high-performance computing
• Open Source: Earlier contributions to open-source projects