Tensor

60 Tensor Jobs Hiring Near You

About the role Own verification of our AI compute core - tensor pipelines, MAC arrays, accumulator logic, and the compute ↔ memory interconnect. Work with chip-design and software teams driving ...

Tensor parallelism and large model scaling * CUDA, NCCL, GPU architecture * GPU partitioning & optimization (MIG) Kubernetes & ML Serving * Kubernetes-based ML serving platforms * KServe, OpenShift ...

AI Core DV Engineer

Mountain View, CA · On-site

$180K - $275K/yr

About the role Own verification of our AI compute core - tensor pipelines, MAC arrays, accumulator logic, and the compute ↔ memory interconnect. Work with chip-design and software teams driving ...

... tensor core acceleration, and compute/memory overlap • Implement cutting-edge features like quantization (FP8/FP4), sparsity, and compute/communication overlap • Identify and resolve performance ...

... tensor core acceleration, and compute/memory overlap • Implement cutting-edge features like quantization (FP8/FP4), sparsity, and compute/communication overlap • Identify and resolve performance ...

Torch, Tensor Flow Experience in Visual SLAM (Simultaneous Localization and Mapping), YOLO, Faster R CNN and deep learning models Knowledge of NVIDIA ISSAC Sim, Omniverse to develop the models Basic ...

Showing results 41-60

Senior Software Engineer, CUTLASS Platform

Nvidia

Austin, TX

$121K - $160K/yr

Full-time

Re-posted 24 days ago


Nvidia rating

9.6

Company rating: 9.6 out of 10

Based on 18 frontline employees who took The Breakroom Quiz

7th of 246 rated software companies


Job description

NVIDIA's high-performance computing platforms are powering the AI revolution across many applications and industries. Within our software stack, CUTLASS stands out as a popular open-source ecosystem dedicated to high-performance linear algebra and Tensor Core primitives. Since 2017, it has provided the community with C++ and Python abstractions to implement custom matrix multiply (GEMM) and related math and deep learning computations on NVIDIA GPUs.

If you are passionate about designing abstractions for Tensor Core and related GPU hardware features in MLIR, Python, and C++ that enable writing high performance kernels, apply to join the CUTLASS team today!

What you'll be doing:

  • Develop core components of the CUTLASS platform including Tensor Core MMAs, copies, synchronization barriers, schedulers, and other GPU hardware features in CUDA C++ and CUTLASS Python DSL.

  • Contribute to the advancement of the MLIR-based backend compiler stack for the CUTLASS Python DSL by designing dialects and associated compiler passes.

  • Author example kernels utilizing CUTLASS abstractions to showcase the use of novel GPU hardware features that are crucial for achieving high performance.

  • Collaborate with GPU architecture, CUDA, and NVVM/PTX compiler teams to provide feedback on programming models and to assess the performance of future GPU hardware features.

What we need to see:

  • Masters or PhD degree in Computer Science, Computer Engineering, or related field (or equivalent experience).

  • 3+ years of relevant industry experience.

  • Strong proficiency in C++ programming and software design, including debugging, performance evaluation, and testing.

  • Experience working with high-performance code generation and knowledge of compiler transformations and optimizations.

  • A deep understanding of computer architecture and parallel computing programming models.

Ways to stand out from the crowd:

  • Experience writing high-performance kernels at low levels of abstractions like NVVM/ PTX for GPUs or other similar parallel processing architectures.

  • Hands-on compiler design experience, particularly in MLIR.

  • Understanding of deep learning models, algorithms, and frameworks.

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hard working people in the world working for us. If you're creative, autonomous, and love a challenge, consider joining our Deep Learning Library team and help us build the real-time, cost-effective computing platform driving our success in this exciting and quickly growing field.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until June 5, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

What Nvidia employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom


Nvidia logo

About Nvidia

Sourced by ZipRecruiter

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology--and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.

Industry

Computer and electronic product manufacturing

Company size

10,000+ Employees

Headquarters location

Santa Clara, CA, US

Year founded

1993