Tensor Jobs (NOW HIRING)

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 ...

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 ...

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 ...

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 ...

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 ...

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 ...

Implement multi-GPU and multi-node model parallelism for serving (tensor or pipeline parallel) * Build autoscaling and load balancing for production ML services * Establish standards for reliability ...

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 ...

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 ...

Responsibilities : • Build, implement, and optimize CUDA kernels for equivariant neural network primitives -- tensor products, segmented polynomials, and triangle-based operations -- targeting peak ...

Lead the definition of mechanisms for efficient movement of tensor activations, weights, and outputs through on-chip and off-chip memory pathways and high-throughput DMA architecture. * Partner ...

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 ...

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 ...

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 ...

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 ...

We're looking for someone who finds joy in memory hierarchies, tensor cores, and profiler output. While San Francisco and Boston are preferred, we are open to other locations. What We're Looking For ...

Lead the definition of mechanisms for efficient movement of tensor activations, weights, and outputs through on-chip and off-chip memory pathways and high-throughput DMA architecture. * Partner ...