About the role Own MLIR dialect design and lowering passes for our AI accelerator - defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML frameworks ...
About the role Own MLIR dialect design and lowering passes for our AI accelerator - defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML frameworks ...
Responsibilities : • Own MLIR dialect design and lowering passes for our AI accelerator -- defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML ...
Responsibilities : • Own MLIR dialect design and lowering passes for our AI accelerator -- defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML ...
Compiler Engineer -- MLIR
Mountain View, CA · On-site
$200K - $360K/yr
Deep understanding of tensor compilation, distributed / sharded execution, and async / streaming dataflow models * Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX ...
Quick apply
Compiler Engineer -- MLIR
Mountain View, CA · On-site
$200K - $360K/yr
Deep understanding of tensor compilation, distributed / sharded execution, and async / streaming dataflow models * Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX ...
Compiler Engineer - MLIR
Mountain View, CA · On-site
$200K - $360K/yr
Deep understanding of tensor compilation, distributed / sharded execution, and async / streaming dataflow models * Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX ...
Compiler Engineer - MLIR
Mountain View, CA · On-site
$200K - $360K/yr
Deep understanding of tensor compilation, distributed / sharded execution, and async / streaming dataflow models * Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX ...
Tensor parallelism and large model scaling * CUDA, NCCL, GPU architecture * GPU partitioning & optimization (MIG) Kubernetes & ML Serving * Kubernetes-based ML serving platforms * KServe, OpenShift ...
Quick apply
Tensor parallelism and large model scaling * CUDA, NCCL, GPU architecture * GPU partitioning & optimization (MIG) Kubernetes & ML Serving * Kubernetes-based ML serving platforms * KServe, OpenShift ...
Responsibilities : • Write and optimize compute kernels for a custom AI accelerator -- tensor operations, data movement patterns, memory hierarchy exploitation • Develop and maintain profiling ...
Responsibilities : • Write and optimize compute kernels for a custom AI accelerator -- tensor operations, data movement patterns, memory hierarchy exploitation • Develop and maintain profiling ...
Write and optimize compute kernels for a custom AI accelerator - tensor operations, data movement patterns, memory hierarchy exploitation * Develop and maintain profiling infrastructure to measure ...
Write and optimize compute kernels for a custom AI accelerator - tensor operations, data movement patterns, memory hierarchy exploitation * Develop and maintain profiling infrastructure to measure ...
Senior Software Engineer - Edge AI/GenAI
San Diego, CA · On-site
$130K - $171K/yr
Strong ability to handle tensor pre-processing and post-processing and integrate AI models into end-to-end pipelines across inputs such as camera, audio, and text. • Strong background in Linux ...
Senior Software Engineer - Edge AI/GenAI
San Diego, CA · On-site
$130K - $171K/yr
Strong ability to handle tensor pre-processing and post-processing and integrate AI models into end-to-end pipelines across inputs such as camera, audio, and text. • Strong background in Linux ...
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 ...
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 ...
Senior Software Engineer - Edge AI/GenAI & Multimedia
San Diego, CA · On-site
$130K - $171K/yr
Strong ability to handle tensor pre-processing and post-processing and integrate AI models into end-to-end pipelines across inputs such as camera, audio, and text. • Strong background in Linux ...
Senior Software Engineer - Edge AI/GenAI & Multimedia
San Diego, CA · On-site
$130K - $171K/yr
Strong ability to handle tensor pre-processing and post-processing and integrate AI models into end-to-end pipelines across inputs such as camera, audio, and text. • Strong background in Linux ...
Principal NPU Microarchitect
Palo Alto, CA · On-site
$218 - $312/hr
At Rivian, the RAP1 SoC compute engine includes a large processing array ( e.g., systolic arrays or tensor cores ), configurable partitioning, instruction DMA engines, data DMA engines, and ...
Principal NPU Microarchitect
Palo Alto, CA · On-site
$218 - $312/hr
At Rivian, the RAP1 SoC compute engine includes a large processing array ( e.g., systolic arrays or tensor cores ), configurable partitioning, instruction DMA engines, data DMA engines, and ...
Kernel Engineer (Compute / Accelerator)
$260K - $320K/yr
Write and optimize compute kernels for a custom AI accelerator -- tensor operations, data movement patterns, memory hierarchy exploitation * Develop and maintain profiling infrastructure to measure ...
Quick apply
Kernel Engineer (Compute / Accelerator)
$260K - $320K/yr
Write and optimize compute kernels for a custom AI accelerator -- tensor operations, data movement patterns, memory hierarchy exploitation * Develop and maintain profiling infrastructure to measure ...
Kernel Engineer (Compute / Accelerator)
Mountain View, CA · On-site
$260K - $320K/yr
Write and optimize compute kernels for a custom AI accelerator - tensor operations, data movement patterns, memory hierarchy exploitation * Develop and maintain profiling infrastructure to measure ...
Kernel Engineer (Compute / Accelerator)
Mountain View, CA · On-site
$260K - $320K/yr
Write and optimize compute kernels for a custom AI accelerator - tensor operations, data movement patterns, memory hierarchy exploitation * Develop and maintain profiling infrastructure to measure ...
Sr. Compiler Engineer, GPU Code Object Rewriting & Tooling
San Jose, CA · On-site
$122K - $168K/yr
Matrix/tensor instructions such as MFMA, WMMA, tensor cores, or other specialized math pipelines. * ROCm profiling tools, hardware performance counters, thread traces, ROCprof, ROCm Compute Profiler ...
Sr. Compiler Engineer, GPU Code Object Rewriting & Tooling
San Jose, CA · On-site
$122K - $168K/yr
Matrix/tensor instructions such as MFMA, WMMA, tensor cores, or other specialized math pipelines. * ROCm profiling tools, hardware performance counters, thread traces, ROCprof, ROCm Compute Profiler ...
Sr. Compiler Engineer, GPU Code Object Rewriting & Tooling
San Jose, CA · On-site
$178K/yr
Matrix/tensor instructions such as MFMA, WMMA, tensor cores, or other specialized math pipelines. * ROCm profiling tools, hardware performance counters, thread traces, ROCprof, ROCm Compute Profiler ...
Sr. Compiler Engineer, GPU Code Object Rewriting & Tooling
San Jose, CA · On-site
$178K/yr
Matrix/tensor instructions such as MFMA, WMMA, tensor cores, or other specialized math pipelines. * ROCm profiling tools, hardware performance counters, thread traces, ROCprof, ROCm Compute Profiler ...
Senior Software Engineer, CUTLASS Kernels
$133K - $175K/yr
Write Tensor Core-based deep learning kernels such as grouped-GEMM, attention, and convolution using CUTLASS CUDA C++ and Python DSL for Blackwell, Rubin, and future architectures. * Optimize kernels ...
Senior Software Engineer, CUTLASS Kernels
$133K - $175K/yr
Write Tensor Core-based deep learning kernels such as grouped-GEMM, attention, and convolution using CUTLASS CUDA C++ and Python DSL for Blackwell, Rubin, and future architectures. * Optimize kernels ...
Senior Software Engineer - Edge AI/GenAI
San Diego, CA · On-site
$130K - $171K/yr
Strong ability to handle tensor pre-processing and post-processing and integrate AI models into end-to-end pipelines across inputs such as camera, audio, and text. * Strong background in Linux system ...
Senior Software Engineer - Edge AI/GenAI
San Diego, CA · On-site
$130K - $171K/yr
Strong ability to handle tensor pre-processing and post-processing and integrate AI models into end-to-end pipelines across inputs such as camera, audio, and text. * Strong background in Linux system ...
Senior Software Engineer -- cuEquivariance
Santa Clara, CA · On-site
$143K - $189K/yr
Responsibilities : • Build, implement, and optimize CUDA kernels for equivariant neural network primitives -- tensor products, segmented polynomials, and triangle-based operations -- targeting peak ...
Senior Software Engineer -- cuEquivariance
Santa Clara, CA · On-site
$143K - $189K/yr
Responsibilities : • Build, implement, and optimize CUDA kernels for equivariant neural network primitives -- tensor products, segmented polynomials, and triangle-based operations -- targeting peak ...
Senior Software Engineer, CUTLASS Platform
Santa Clara, CA · On-site
$143K - $189K/yr
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 ...
Senior Software Engineer, CUTLASS Platform
Santa Clara, CA · On-site
$143K - $189K/yr
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 ...
Senior Software Engineer, CUTLASS Kernels
Santa Clara, CA · On-site
$143K - $189K/yr
Write Tensor Core-based deep learning kernels such as grouped-GEMM, attention, and convolution using CUTLASS CUDA C++ and Python DSL for Blackwell, Rubin, and future architectures. * Optimize kernels ...
Senior Software Engineer, CUTLASS Kernels
Santa Clara, CA · On-site
$143K - $189K/yr
Write Tensor Core-based deep learning kernels such as grouped-GEMM, attention, and convolution using CUTLASS CUDA C++ and Python DSL for Blackwell, Rubin, and future architectures. * Optimize kernels ...
Tensor information
See salary details
$46K - $64K
1% of jobs
$64K - $81.9K
2% of jobs
$81.9K - $99.9K
4% of jobs
$99.9K - $117.8K
9% of jobs
$133.1K is the 25th percentile. Wages below this are outliers.
$117.8K - $135.8K
11% of jobs
$135.8K - $153.7K
7% of jobs
The median wage is $159.5K / yr.
$153.7K - $171.7K
50% of jobs
$171.7K - $189.6K
2% of jobs
$189.6K - $207.6K
1% of jobs
$207.6K - $225.5K
0% of jobs
$225.5K - $243.5K
13% of jobs
$46K
$165K
$243.5K
How much do tensor jobs pay per year?
What are Tensor jobs?
What is the best job for an Overthinker?
What is a Tensor job?
A Tensor job typically refers to a role involving tensors, which are mathematical objects used in machine learning, AI, and scientific computing. These jobs often require expertise in deep learning frameworks like TensorFlow or PyTorch, where tensors represent multi-dimensional arrays for data processing. A Tensor job can involve designing and optimizing neural networks, performing large-scale data analysis, or working with high-performance computing.
What is the difference between Tensor vs Data Scientist?
| Aspect | Tensor | Data Scientist |
|---|---|---|
| Required Credentials | Knowledge of machine learning, programming skills, often a degree in computer science or related fields | Degree in statistics, computer science, or related fields; strong analytical skills |
| Work Environment | Tech companies, AI research labs, software development teams | Business, finance, healthcare, and tech industries analyzing data to inform decisions |
| Industry Usage | Primarily in AI, machine learning, and deep learning projects | Across industries for data analysis, predictive modeling, and insights |
While a Tensor is a fundamental data structure used in machine learning frameworks like TensorFlow, a Data Scientist analyzes data to extract insights and build models. Tensors are tools that Data Scientists often work with, but they are not roles themselves. Understanding tensors is essential for Data Scientists involved in AI and machine learning projects.
What are some common challenges faced by TensorFlow Developers when working on large-scale machine learning projects?
What are the key skills and qualifications needed to thrive as a Tensor?
What jobs use TensorFlow?
What jobs pay 4000 a week without a degree?
Which 3 jobs will survive AI?

Other
Medical, Dental, Vision, Retirement, PTO
Re-posted 28 days ago
Job description
Own MLIR dialect design and lowering passes for our AI accelerator - defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML frameworks to silicon. Work with chip-design and software teams driving DensityAI's AI accelerator program from first silicon through scale-out.
What you'll do- Own MLIR dialect design and lowering passes for our AI accelerator - defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML frameworks to silicon
- Use and develop AI-assisted tool flows to accelerate compiler development
- Exceptional abilities in MLIR dialect design, lowering pass authoring, and rewrite patterns
- 5+ years compiler engineering experience, with hands-on MLIR contributions or equivalent IR-design experience
- Deep understanding of tensor compilation, distributed / sharded execution, and async / streaming dataflow models
- Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX, TensorFlow, or equivalent)
- (Optional) LLVM backend experience, GPU compiler experience (Triton, IREE, XLA, or equivalent), or open-source MLIR / LLVM contributions
Final offers depend on level, location, and skills relevant to the role. Additional compensation: equity grant per company guidelines; medical / dental / vision; 401(k); standard PTO.
Visa SponsorshipDensityAI sponsors qualified candidates for H-1B, O-1, TN, E-3, and other employment-based visas, and we welcome applicants on F-1 OPT and STEM-OPT. Work authorization is required at start; we provide immigration support to secure or transfer status.
Equal OpportunityDensityAI is an Equal Opportunity Employer. We do not discriminate on the basis of race, color, religious creed, national origin, ancestry, physical or mental disability, medical condition, genetic information, marital status, sex, gender, gender identity, gender expression, age (40+), sexual orientation, military or veteran status, pregnancy, or any other status protected by law. We comply with the California CROWN Act and provide reasonable accommodations on request.