@@ -90,7 +90,7 @@ numr implements a comprehensive set of tensor operations across CPU, CUDA, and W
9090### Shape and Data Movement
9191
9292- ** ShapeOps** : cat, stack, split, chunk, repeat, pad, roll
93- - ** IndexingOps** : gather, scatter, gather_nd, scatter_reduce, index_select, masked_select, masked_fill, embedding_lookup, bincount, argmax, argmin
93+ - ** IndexingOps** : gather, scatter, gather_nd, scatter_reduce, index_select, masked_select, masked_fill, embedding_lookup, bincount, argmax, argmin, slice_assign
9494- ** SortingOps** : sort, argsort, topk, unique, nonzero, searchsorted
9595
9696### Reductions
@@ -106,16 +106,18 @@ numr implements a comprehensive set of tensor operations across CPU, CUDA, and W
106106
107107### Activation & Normalization Functions
108108
109- - ** ActivationOps** : relu, sigmoid, silu, gelu, swiglu, leaky_relu, elu, softmax, dropout
110- - ** NormalizationOps** : rms_norm, layer_norm, batch_norm, group_norm, instance_norm
109+ - ** ActivationOps** : relu, sigmoid, silu, gelu, swiglu, leaky_relu, elu, softmax, dropout, fused activation-mul (for gated architectures)
110+ - ** NormalizationOps** : rms_norm, layer_norm, batch_norm, group_norm, instance_norm, fused add-norm (residual + normalize in one pass)
111+ - ** GemmEpilogueOps** : fused matmul+bias+activation in a single kernel (forward + backward)
112+ - ** FusedElementwiseOps** : fused element-wise operation chains across all backends
111113- ** ConvOps** : conv1d, conv2d, depthwise_conv2d (with stride, padding, dilation, groups)
112114- ** EinsumOps** : Einstein summation notation
113115
114116_ These are mathematical functions commonly used in ML, but numr itself is not an ML framework._
115117
116118### Linear Algebra
117119
118- - ** MatmulOps** : matmul, matmul_bias (fused GEMM+bias)
120+ - ** MatmulOps** : matmul, matmul_bias (fused GEMM+bias), i8×i8→i32 quantized matmul, FP8 matmul
119121- ** LinalgOps** : solve, lstsq, pinverse, inverse, det, trace, matrix_rank, diag, matrix_norm, kron, khatri_rao
120122- ** ComplexOps** : conj, real, imag, angle (for complex tensor support)
121123
@@ -126,11 +128,12 @@ _These are mathematical functions commonly used in ML, but numr itself is not an
126128- ** Second-order** : ` hvp() ` for Hessian-vector products, ` backward_with_graph() ` for higher-order gradients
127129- ** Activation checkpointing** : ` checkpoint() ` to trade compute for memory
128130- ** Backward hooks** : ` BackwardHook ` trait for gradient notifications (e.g., distributed allreduce)
131+ - ** Differentiable ops** : matmul, conv1d, conv2d, softmax, rms_norm, layer_norm, SiLU, softplus, SwiGLU, dropout, fused GEMM epilogue, fused add-norm, dtype cast, narrow, cat
129132
130133### Statistics and Probability
131134
132135- ** StatisticalOps** : var, std, skew, kurtosis, quantile, percentile, median, cov, corrcoef
133- - ** RandomOps** : rand, randn, randint, multinomial, bernoulli, poisson, binomial, beta, gamma, exponential, chi_squared, student_t, f_distribution
136+ - ** RandomOps** : rand, randn, randint, multinomial, bernoulli, poisson, binomial, beta, gamma, exponential, chi_squared, student_t, f_distribution (with seeded deterministic generation)
134137- ** MultivariateRandomOps** : multivariate_normal, wishart, dirichlet
135138- ** QuasirandomOps** : Sobol, Halton sequences
136139
@@ -185,6 +188,7 @@ _These are mathematical functions commonly used in ML, but numr itself is not an
185188
186189- Formats: CSR, CSC, COO
187190- Operations: SpGEMM (sparse matrix multiplication), SpMV (sparse matrix-vector), DSMM (dense-sparse matrix)
191+ - 2:4 structured sparsity with multi-backend support
188192
189193** Sparse Linear Algebra (` numr::sparse_linalg ` ):**
190194
@@ -234,15 +238,15 @@ Every operation supports every compatible dtype. No hardcoded f32-only kernels.
234238
235239All backends implement identical algorithms with native kernels—no cuBLAS, MKL, or vendor library dependencies.
236240
237- | Hardware | Backend | Feature | Status | Notes |
238- | ------------ | ------- | ------------- | ------- | ------------------ |
239- | CPU (x86-64) | CPU | cpu (default) | ✓ | AVX-512/AVX2 SIMD |
240- | CPU (ARM64) | CPU | cpu | ✓ | NEON SIMD |
241- | NVIDIA GPU | CUDA | cuda | ✓ | Native PTX kernels |
242- | AMD GPU | WebGPU | wgpu | ✓ | WGSL shaders |
243- | Intel GPU | WebGPU | wgpu | ✓ | WGSL shaders |
244- | Apple GPU | WebGPU | wgpu | ✓ | WGSL shaders |
245- | AMD GPU | ROCm | - | Planned | Native HIP kernels |
241+ | Hardware | Backend | Feature | Status | Notes |
242+ | ------------ | ------- | ------------- | ------- | ------------------------------------------------------ |
243+ | CPU (x86-64) | CPU | cpu (default) | ✓ | AVX-512/AVX2 SIMD |
244+ | CPU (ARM64) | CPU | cpu | ✓ | NEON SIMD |
245+ | NVIDIA GPU | CUDA | cuda | ✓ | Native PTX kernels, caching allocator, GEMV fast paths |
246+ | AMD GPU | WebGPU | wgpu | ✓ | WGSL shaders |
247+ | Intel GPU | WebGPU | wgpu | ✓ | WGSL shaders |
248+ | Apple GPU | WebGPU | wgpu | ✓ | WGSL shaders |
249+ | AMD GPU | ROCm | - | Planned | Native HIP kernels |
246250
247251### SIMD Acceleration
248252
0 commit comments