High-performance GPU kernels for LLM inference in OpenAI Triton. Fused RMSNorm, SwiGLU, INT8 GEMM with benchmarks and roofline analysis.
-
Updated
Jul 22, 2026 - Python
High-performance GPU kernels for LLM inference in OpenAI Triton. Fused RMSNorm, SwiGLU, INT8 GEMM with benchmarks and roofline analysis.
JAX benchmarking, profiling and evaluation metrics for Flax NNX: a registry of pure-function metrics (regression, classification, calibration, uncertainty, forecasting, generative, image, text, audio, graph, fairness), XLA FLOP counting, roofline analysis, GPU and energy monitoring, regression detection, publication exports, W&B and MLflow.
Kernel-level profiling of batch-1 decode on consumer GPU to prove that decode is memory-bandwidth-bound against the hardware roofline, then beating the baseline with a fused dequant+GEMV CUDA kernel. (regime-aware attribution at the end)
ARM NEON SIMD optimizations on Apple Silicon for attention (31.88 GFLOPS/s, 11.1x speedup) and Black-Scholes (1.03x, proving SIMD fails on transcendental-heavy workloads). Roofline analysis predicts optimization potential before implementation. OpenMP parallelization.
Three measured notebooks on GPU inference optimization: roofline analysis, KV-cache decode optimization (4.21x), and CUDA-graph launch-overhead elimination (5.38x). Pure PyTorch.
High-performance computing project exploring CPU and GPU matrix multiplication optimization through OpenMP, CUDA, tiled kernels, and cuBLAS, with detailed performance analysis across large matrix sizes
To associate your repository with the roofline-analysis topic, visit your repo's landing page and select "manage topics."