Math Device Extension Libraries Intern - 2027 — NVIDIA (ufficio Zurich)
- Location
- Zürich
- Contract
- Full-time
- Posted
Full description
We are looking for software engineering interns to help us expand our catalog of Device eXtension (Dx) APIs for our math libraries. For over a decade, NVIDIA's accelerated computing platform has revolutionized HPC and AI with applications ranging from COVID-19 research to autonomous machines. Our team develops the GPU-accelerated mathematical libraries that make all of this possible. The hardware and software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR…), and optimization at runtime for maximum flexibility and performance. Our Dx APIs allow developers to embed highly-optimized mathematical operations in their applications for these and other scenarios. You will be part of a team designing, developing, and optimizing math libraries for the future. Does the idea of being at the heart of these projects and applying your knowledge to develop and optimize algorithms which make an impact around world excite you? If yes, then come and join our team!
During your internship, you will work with senior software engineers on the libraries team who will provide mentorship and guide you in developing highly optimized algorithms. Projects will involve implementing new numerical algorithms, defining APIs, analyzing performance, performance optimizations on the latest NVIDIA GPUs, improving libraries customer experience in employing agentic workflows and AI tools for their development and other general software engineering work.
What you'll be doing
- Prototype and develop numerical algorithms for high-performance math device extension libraries: cuBLASDx, cuSolverDx, cuFFTDx, cuRANDDx, nvCOMPDx, NPPDx or libmathdx on ground breaking NVIDIA graphics processors.
- Analyze the performance of GPU implementations and find opportunities for improvements
- Collaborate with team members to understand software use cases and requirements
- Prototype tooling and solutions to help transform math libraries’ APIs into LLM-friendly representations across popular codegen tools.
What we need to see
- Pursuing a MS or PhD degree in Computer Science, Artificial Intelligence, Applied Mathematics, 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.
Ways to stand out from the crowd
- Understanding of composability and fusions, compilers, implementation of programming languages
- Knowledge of GPU and network hardware architecture
- Familiarity with Python and recent CUDA developments like cuTile and CUDA Tile C++
- Prior work in AI-assisted software development, knowledge of LLMs, MCP, and building RAGs or agenting tooling
- Exposure to numerical methods, floating-point arithmetic and numerical error analysis
We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you! We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/
Questions about this listing
What salary does NVIDIA (ufficio Zurich) offer for this role?
The original listing does not state a salary. Our estimate for this role and sector in Zürich is CHF 62'000 - 94'000 gross per year — an estimate, not a figure communicated by NVIDIA (ufficio Zurich). The calculator on this site converts a gross salary into your actual net take-home once cross-border tax and social contributions are applied.
Is this a full-time role, and what type of contract does NVIDIA (ufficio Zurich) offer?
This listing is a full-time position. The contract type shown here comes directly from the employer's original posting; always confirm exact hours, notice period and probation length with NVIDIA (ufficio Zurich) during the application process, since these details can vary by role even within the same contract category.
Do I need a cross-border work permit for a role in Zürich?
EU/EFTA citizens living in an EU/EFTA state can apply for a G permit and can work anywhere in Switzerland: border zones no longer apply to them, and they must return home at least once a week. The Swiss employer files the request with Canton Zürich's migration office after the contract is signed. Processing times vary by canton, so confirm the specifics with Zürich's cantonal migration office or with HR during the application.
How do I apply for this position at NVIDIA (ufficio Zurich)?
Use the "Apply now" button on this page — it links directly to NVIDIA (ufficio Zurich)'s original listing at nvidia.wd5.myworkdayjobs.com, so your application goes straight to the employer's own applicant-tracking system. Frontaliere Ticino does not collect or forward applications itself.
