Tensor Jobs in California (NOW HIRING)

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

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

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

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

Optimize edge inference performance: model quantization (INT8/FP16), Tensor RT engine compilation, DLA offloading, and latency profiling to meet real-time frame rate targets under concurrent multi ...

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

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

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

Tensor flow and Deep Learning hands on experience * LLM Hands on experience * Very good analytical skills and communication skills and Directly discuss with client and sending the status reports.

... tensor cores, to improve compute efficiency LLM Inference Internals · Optimize prefill and decode stages, attention mechanisms, and continuous batching · Implement and tune quantization ...

... tensor cores, to improve compute efficiency • Optimize prefill and decode stages, attention mechanisms, and continuous batching • Implement and tune quantization, speculative decoding, tensor ...

The Google Security Architecture (GSA) is a security element in every Pixel phone's Tensor SoC. It is an independent CPU and separate from the main Application Processor, but can access the main bus ...

FSDP/ZeRO/tensor+pipeline parallel; NCCL tuning. GPU + kernel performance * Nsight profiling, Triton/CUDA kernels, fused ops. * Flash-attention-style speedups, sequence packing, KV-cache tricks.

Your work will span the entire stack-from low-level tensor core optimizations to orchestrating thousands of GPUs in perfect synchronization. Strong candidates will have a track record of delivering ...

You will have an opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology that powers Google's most demanding AI/ML applications. You'll be part of a team that pushes boundaries ...

Strong experience with AI and GenAI inference frameworks including Py-Torch, TensorFlow, ONNX Runtime, and Llama CPP, with solid understanding of AI fundamentals, model architectures, tensor layouts ...

Deep expertise in 3D parallelism (Data, Tensor, Pipeline). * Experience managing SLURM or Kubernetes-based GPU clusters. * Strong systems engineering background (C++, CUDA, Python)