GPU Math Libraries Intern: Optimize Dx APIs

NVIDIA Corporation; Warszawa; On-site; PLN 118,000 - 204,000; Full time

Job summary

NVIDIA is seeking software engineering interns to help expand our Dx API catalog for math libraries on NVIDIA GPUs. You will work with senior engineers on implementing and optimizing high‑performance algorithms for cuBLASDx, cuSolverDx, and related libraries.

You will learn about GPU architectures, performance analysis, and API design while contributing to ground‑breaking math software. This internship is based in Poland, Warsaw, with mentorship and exposure to real-world projects in AI and HPC.

Qualifications

• Pursuing MS/PhD in CS, AI, Applied Math, or related field.
• Strong C++ programming, debugging, performance analysis, and test building skills.
• Parallel or GPU programming experience using CUDA/OpenMP/OpenCL/SYCL or similar.
• Experience with linear algebra or FFT or data compression algorithms.
• Collaboration, communication, and documentation habits.

Responsibilities

• Prototype and develop numerical algorithms for high-performance math device extension libraries: cuBLASDx, cuSolverDx, cuFFTDx, cuRANDDx, nvCOMPDx, NPPDx or libmathdx.
• Analyze the performance of GPU implementations and find opportunities for improvements.
• Collaborate with team members to understand software use cases and requirements.
• Prototype tooling to transform math libraries' APIs into LLM-friendly representations across codegen tools.

Skills

C++ programming

Performance analysis

Debugging

Test building

Parallel/GPU programming

Mathematics background

Communication skills

Education

MS or PhD candidate in CS/AI/Applied Math

Tools

CUDA toolkit

Python basics

Job description

NVIDIA is seeking software engineering interns to help expand our Dx API catalog for math libraries on NVIDIA GPUs. You will work with senior engineers on implementing and optimizing high‑performance algorithms for cuBLASDx, cuSolverDx, and related libraries.

You will learn about GPU architectures, performance analysis, and API design while contributing to ground‑breaking math software. This internship is based in Poland, Warsaw, with mentorship and exposure to real-world projects in AI and HPC.