Senior AI Frameworks Engineer

Nvidia

Actively hiring
Santa Clara, US · Austin, US Posted 24 days ago $152,000–$241,500 / year
View original post Log in to save

At a glance

AI generated

TL;DR

Join NVIDIA’s CUTLASS team as a Senior AI Frameworks Engineer and contribute to the development of Pythonic CUTLASS, an initiative aimed at integrating high-performance math primitives into the Python environment. You will design APIs that enhance developer productivity while maintaining performance, develop robust compilation infrastructure for efficient GPU code generation from Python descriptions, and create debugging tools and profilers to optimize user experience. Additionally, you will manage package distribution and documentation for the open-source community. Ideal candidates have a strong background in Python and C++, with expertise in library development, intuitive API design, and knowledge of compiler foundations such as IR and AST manipulation. Experience with GPU architecture and parallel programming models is also essential.

Skills

Python C++ CUDA LLVM MLIR AST JIT FFI CI/CD GitHub Conda Wheels Documentation Testing OpenSource GPUProgramming

What you'll do

  • Design APIs that enhance user productivity for modern scientific computing frameworks.
  • Develop robust compilation infrastructure to transform Python descriptions into high-performance GPU code.
  • Create debugging tools and profiler integrations to simplify kernel development and usage.
  • Build production-grade delivery infrastructure including package distribution and documentation.
  • Optimize developer experience with validation methodologies and testing for CUTLASS.

What we're looking for

  • MS or PhD in Computer Science, Electrical Engineering, or related field required.
  • At least 3+ years of experience in relevant technical roles.
  • Strong proficiency in Python and C++ for designing extensions and FFI.
  • Experience in developing high-performance open-source libraries or frameworks.
  • Deep understanding of the Python ecosystem’s delivery stack and tools.
  • Active maintainer status or significant contributions to AI frameworks or compilers.

Market check

Salary context

This $152,000–$241,500 range sits above 39% of similar postings on FindRole.

Peer median band

$178,875–$260,295

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$174,600–$246,150

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 801 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 797 roles with salary data.

Most-posted roles

View all roles at Nvidia

More like this

Similar roles

Senior AI Solutions Engineer

Fiserv

Berkeley Heights, New Jersey, US 14 days ago $128,000–$216,000
Python Typescript JavaScript LLM APIs Claude Code LangChain Anthropic's agent SDK Semantic Kernel AutoGen CrewAI M365 Copilot Copilot Studio Power Platform Microsoft Graph Azure AI services RAG architectures Vector databases Embedding models Agent evaluation frameworks LLM observability

Senior AI Solutions Engineer

Elevance Health

US 11 days ago $132,088–$198,132
Python SQL AWS Bedrock LLMs OpenSearch RAG NLP APIs MLOps CI/CD Healthcare data privacy Responsible AI principles Cloud-based ML platforms Microservices Evaluation frameworks for ML/LLM systems

Senior AI Platform Engineer

Adobe

San Jose, US 74 days ago $211,800–$306,625
TypeScript Python Java Go C++ LLMs Terraform Kubernetes Docker CI/CD Prometheus Grafana PostgreSQL Redis Elasticsearch AWS Azure Google Cloud Platform Git Jenkins GitHub Slack Confluence Jira Swagger OpenAPI GraphQL RESTful APIs Microservices MVP

Senior Applied AI Engineer

Abbott

Remote (US) 142 days ago $86,700–$173,300
Python Java C# AWS Azure NLP Generative AI Docker Kubernetes CI/CD PostgreSQL MongoDB Git Jenkins
Remote

Senior Applied AI Engineer

Equifax

Usa - Georgia - Alpharetta - 30005, US 18 days ago
GCP Kubernetes Terraform Java Spring Boot Apache Beam Bigtable BigQuery PubSub GCS Composer Airflow Jenkins Helm CI/CD Python SQL NLP LLM CRMs APIs Event-driven architectures MLOps Docker Prometheus Grafana

Senior AI Engineer

Adobe

San Jose, US 57 days ago $187,100–$270,950
Python Azure AWS GCP Spark Databricks LLMs RAG AI-assisted coding tools CI/CD GitHub Copilot Claude Code Cursor Prometheus Grafana