MatX represents a series b bet on horizontal AI tooling, with enhancement GenAI integration across its product surface.
MatX enters a market characterized by significant capital deployment and growing enterprise adoption. The current funding environment favors companies with clear technical differentiation and defensible market positions.
MatX is an AI semiconductor company that designs custom chips and hardware architectures to support large language models.
Integrated co-design of silicon, low‑precision compute formats, compiler/kernels and novel attention algorithms (MXFP4 + training-safe block-quantized attention + SMVA/Sparse V + SD-aware sparse attention) that together reduce memory bandwidth and increase throughput for LLM decoding and training.
MatX uses speculative decoding which explicitly employs multiple models (a lightweight draft model and a heavier target model) working together to accelerate decoding. This is an ensemble/mesh pattern where work is split across cheaper and more expensive models with coordinated verification.
Cost-effective AI deployment for mid-market. Creates opportunity for specialized model providers.
MatX builds on Llama family, Leaky (1B-parameter model variant), Fixed (1B-parameter model variant), leveraging MatX (in-house) infrastructure with JAX, Seqax codebase in the stack. The technical approach emphasizes hybrid.
Speculative decoding: a draft model generates candidates; a target model verifies. They modify attention selection so draft blocks reuse the same KV block selection to preserve sparsity during verification.
Head-level routing: decouple number of K heads and V heads (MVA) and partition heads based on gcd; no evidence of dynamic model selection or higher-level multi-model routing beyond speculative decoding's draft/target.
Over 28 years in hardware architecture and ML accelerators; Efficiency Lead for Google PaLM; designed and implemented the world's fastest LLM inference software; helped conceive Google's TPU v5e; Architect and Compiler Lead for Google's ML chips.
Previously: Google
Over 25 years building products and leading engineering teams in silicon and software; Founding team member at Innovium, contributed to a $1.1B exit; led multiple chip products from concept to high-volume deployment.
Previously: Innovium
high
developer first
Target: developer
hybrid
Research and development of efficient attention mechanisms for training/inference in transformers to increase throughput and reduce memory bandwidth
This is a concrete, surgical mitigation of a subtle correctness issue introduced by block quantization, enabling end-to-end training with attention quantized (including gradients). It combines low-level numeric choices (stochastic rounding) with algorithmic selective dequantization — uncommon in standard quantization work.
Decoupling K and V head counts is a practical, quantifiable tradeoff between memory bandwidth and quality; combining it with training-time sparsity (threshold-based) to achieve large sparsity ratios is uncommon and impactful for inference bandwidth-limited deployments.
Shifting sparsification into training and using a cheap threshold makes sparse V practical and efficient, avoids top-k costs, and achieves much higher sparsity for same quality — a practical systems+training co-design.
MatX operates in a competitive landscape that includes NVIDIA, Google (TPU / Vertex / TPU Research Cloud), Cerebras.
Differentiation: MatX emphasizes custom silicon + co-designed compilers/kernels optimized for very low-precision compute (MXFP4-style methods and block-quantized attention training fixes) and claims an end-to-end hardware+software co-design focus rather than general-purpose GPU-first approach.
Differentiation: MatX positions itself as a smaller, faster-moving team that bets heavily on extreme low-precision compute and in-house ML research (seqax, MXFP4, attention quantization fixes) targeted at LLM inference/training efficiency and KV-cache bandwidth reductions.
Differentiation: MatX focuses on block-quantized formats, low-precision microformats, and algorithmic changes (SMVA, Sparse V, MXFP4-safe training) to reduce memory bandwidth and enable efficiency on modern accelerators; MatX stresses tighter integration of quantized training techniques with chip design.
They discovered a subtle causal leak introduced by block-wise quantization: the per-block shared scale (used in microscaling/MXFP4) makes quantized K/V at earlier token positions depend on later tokens inside the same quantization block, which breaks causal LM training.
Their surgical fix: keep the specific block-diagonal tile multiply (the tile where query-block index == key-block index) unquantized while using block-quantized matmuls everywhere else. This preserves throughput for most ops but blocks the single leakage path.
They quantize not only forward attention matmuls but also the attention gradient, and use different rounding strategies: round-to-nearest for forward, stochastic rounding for backward—an explicit rounding-mode design to stabilize low-precision training.
Novel experimental sanity check: compare parallel prefill loss vs autoregressive (one-token-at-a-time) loss to detect future leakage from quantization. The gap served as an operational signal for leakage.
They decouple K and V head counts (Multi Value Attention, MVA): use one or few K heads but many V heads. This recognizes that V heads drive quality while K heads dominate KV cache bandwidth.
If MatX achieves its technical roadmap, it could become foundational infrastructure for the next generation of AI applications. Success here would accelerate the timeline for downstream companies to build reliable, production-grade AI products. Failure or pivot would signal continued fragmentation in the AI tooling landscape.
“Quantizing attention improves efficiency on two fronts: the model has higher compute throughput, and loads fewer bytes per key/value.”
“training with MXFP4 in both attention and the attention gradient”
“We trained a pair of 1B-parameter models on the C4 dataset, with MXFP4 for attention and the attention gradient.”
“The Fixed model remained well behaved throughout training. However, the Leaky model had training dynamics associated with future leakage.”
“During experiments, we compared attention mechanisms such as GQA, MQA, MVA, SMVA and MHA, with specific references to their FLOPs, KV cache requirements, and losses.”
“We trained 101 million parameter model using the seqax codebase with various attention variants.”