Senior Staff LLM Inference Engineer
D Matrix
Santa Clara, CA · $195k–$285k a year · Posted August 18, 2026
Just laid off? Start here: a plan for the first weeks
See every open role at D Matrix
More in California job market · Technology & Software · Engineering jobs
More roles at D Matrix
- Director, Quality and Reliability Engineering – AI Hardware · Santa Clara, CA
- Sr. Staff Power Performance Architect · Santa Clara, CA
- Director of Sales · Seattle, WA
- Principal Software Engineer, SDK & Lowering Stack · Santa Clara, CA
- Staff Analog Design Engineer, SerDes Clocking · Santa Clara, CA
- Sr. Staff Thermal Design Engineer · Santa Clara, CA
Similar jobs
- Senior Engineer, Systems · Sonova USA · Valencia, CA
- Senior Engineer, Systems - Embedded Electronics · Sonova USA · Valencia, CA
- Sr. Engineer, Aerodynamics · Virgin Galactic Holdings · Tustin, CA
- SMTS Software Development Engineer · Advanced Micro Devices · Santa Clara, CA
- Sr. Silicon Design Engineer · Advanced Micro Devices · Santa Clara, CA
- Senior Technical Program Manager, Engineering Strategy & Business Operations · Keysight Technologies · Santa Rosa, CA
Where Santa Clara, CA ranks
Where Santa Clara, CA placed in our monthly rankings, month by month. Each month stays online at its own address.
November 2026
- #1Best cities for sales pay after cost of living$175k · of 45
- #1Best cities for product pay after cost of living$224k · of 14
- #2Best cities for operations pay after cost of living$173k · of 41
- #2Best cities for marketing pay after cost of living$192k · of 16
- #2Best all-round cities for remote workers in the West Coast85 / 100 · of 39
- #3Best cities for finance pay after cost of living$156k · of 35
Rankings about engineering jobs
- Best cities for software and engineering jobs #1 San Francisco (3,412 openings)
- Best cities for engineering pay after cost of living #1 San Mateo ($216k)
- Most requested AI skills in engineering jobs #1 AI tools in daily work (6.1%)
- AI tools named most often in engineering jobs #1 Claude (1.7%)
- Employers with the most engineering jobs #1 Amazon (2,992 roles)
Skills this job asks for
Say which you have and which you want to build. We use it to match you to jobs like this one.
About the job
Join d-Matrix and help build the future of AI inference. We’ve pioneered a memory-centric architecture that brings compute directly into memory, making AI faster, more energy efficient, and accessible on a global scale. From chips and software to rack-scale systems, our team is tackling some of AI infrastructure’s hardest challenges — and shipping breakthrough technology today. If you want to create what comes next and help make infinite inference possible, we’d love to hear from you.
Our culture is built on respect, collaboration, and direct communication. We value humility, we believe our different perspectives lead to better solutions, and we’re looking for people who are driven by execution and energized by hard problems.
D-Matrix Frontier Group sits at the leading edge of what’s possible with LLM inference on heterogeneous hardware. Our charter spans the full stack: from pathfinding emerging use cases and novel deployment patterns to deep optimization of inference kernels to building proof-of-concept systems that showcase D-Matrix’s unique computational fabric. We are an applied research and engineering team that moves fast, ships real systems, and works directly with product and hardware teams to shape the roadmap.
We build the tools, runtimes, and frameworks that let frontier AI models run efficiently and cost-effectively across heterogeneous deployments combining D-Matrix silicon with CPUs, GPUs, and custom accelerators. Our work powers everything from benchmarking and evaluation pipelines to production-grade inference serving.
This Role
We are hiring end-to-end inference engineers who are comfortable going from a novel research idea to a deployed, optimized system. You will work at every layer of the inference stack from kernel-level optimization to distributed orchestration to high-level serving APIs.
This role could be a great match for you if you:
- Have deep intuition for modern generative AI architectures and how to squeeze performance out of them at inference time.
- Are familiar with the internals of open-source inference frameworks (vLLM, SGLang, TensorRT-LLM, etc.) and can extend or replace them when needed.
- Enjoy pathfinding new use cases — exploring heterogeneous deployment topologies and building early-stage POCs that prove out new ideas.
- Are results-oriented with a strong bias toward action; you own problems end-to-end from prototype to optimization to handoff.
- Are energized by working at the intersection of novel hardware and frontier models and want your work to directly influence how next-generation AI silicon is used.
- Value clear communication and thrive in a small, high-ownership team environment.
Responsibilities
- Identify and prototype emerging LLM inference use cases suited to heterogeneous hardware deployments.
- Build compelling proof-of-concept systems that demonstrate D-Matrix capabilities to customers, partners, and internal stakeholders.
- Develop and tune custom kernels and operator-level optimizations to maximize throughput and minimize latency.
- Drive quantization, sparsity, and batching strategies tailored to the D-Matrix computational model.
- Build and maintain inference runtimes, serving frameworks, and evaluation tooling.
- Contribute to distributed inference systems: tensor/pipeline parallelism, disaggregated prefill/decode, KV-cache management.
- Work closely with hardware architects to provide firmware and compiler teams with actionable inference workload insights.
- Partner with product and business development to translate POCs into customer-facing demonstrations.
- Contribute to technical publications, whitepapers, and open-source projects that advance D-Matrix visibility.
Required Qualifications
- Bachelor’s degree in Computer Science, Electrical Engineering, or a related field, and 10+ years of relevant engineering experience or equivalent demonstrated experience.
- Master’s or PhD in Computer Science, Electrical Engineering, or a related field preferred, with 6+ years of relevant industry experience.
- Strong proficiency in Python and C/C++.
- Hands-on experience optimizing LLM inference — attention kernels, KV cache, batching strategies, quantization (INT8/FP8/INT4).
- Experience with at least one major inference framework (vLLM, SGLang, TensorRT-LLM, ONNX Runtime, or similar) at a contributor level.
- Familiarity with GPU kernel programming (CUDA/Triton) and performance profiling tools.
Preferred Qualifications
- Experience with heterogeneous compute deployments — scheduling inference workloads across dissimilar hardware (accelerators, CPUs, GPUs).
- Familiarity with custom silicon or ASIC-based inference (beyond GPU-only environments).
- Experience with distributed inference: tensor parallelism, pipeline parallelism, disaggregated serving.
- Contributions to open-source inference or ML systems projects.
- Experience with production inference serving at scale (latency SLOs, continuous batching, multi-model serving).
- Familiarity with speculative decoding, mixture-of-experts routing, or long-context-serving techniques.
- Working familiarity with the material in the JAX Scaling Book or equivalent systems-level understanding of modern LLM training and inference.
Why D-Matrix Frontier Group
- Work on genuinely novel hardware: D-Matrix in-memory compute architecture opens up inference optimization problems that don’t exist anywhere else.
- End-to-end ownership from idea to deployed system, with a short feedback loop between your work and real hardware.
- Small, senior team with high autonomy and direct influence on product direction.
- Competitive compensation, equity, and benefits in Santa Clara, CA.
Equal Opportunity Employment Policy
d-Matrix is proud to be an equal opportunity workplace and affirmative action employer. We’re committed to fostering an inclusive environment where everyone feels welcomed and empowered to do their best work. We hire the best talent for our teams, regardless of race, religion, color, age, disability, sex, gender identity, sexual orientation, ancestry, genetic information, marital status, national origin, political affiliation, or veteran status. Our focus is on hiring teammates with humble expertise, kindness, dedication, and a willingness to embrace challenges and learn together every day.
d-Matrix does not accept resumes or candidate submissions from external agencies. We appreciate the interest and effort of recruitment firms, but we kindly request that individuals interested in opportunities with d-Matrix apply directly through our official channels. This approach allows us to streamline our hiring processes and maintain a consistent and fair evaluation of all applicants. Thank you for your understanding and cooperation.