Skip to content
View Mattral's full-sized avatar
👀
I may be slow to respond.
👀
I may be slow to respond.

Highlights

  • Pro

Block or report Mattral

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Maximum 250 characters. Please don’t include any personal information such as legal names or email addresses. Markdown is supported. This note will only be visible to you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
mattral/README.md

Mattral

ML Systems Engineer · Distributed Training · LLM Infrastructure · Multimodal Systems

I build systems that work at scale -- and try to understand why they fail when they do.


What I actually do

I work in the space between clean research ideas and the messy reality of clusters that fail, data that drifts, and models that need to stay honest in production.

Day-to-day: cloud-scale ML infrastructure at a hyperscaler, distributed training systems, fault-tolerant checkpointing, LLM safety and observability layers, and the occasional low-level kernel when something needs to be faster or more reliable. The majority of that work lives in private repositories. What you see here are the side projects I chose to open-source because they felt worth sharing.

Things I care about technically

  • Large-scale pre-training infrastructure -- MoE routing, fault-tolerant checkpointing, tensor/pipeline parallelism
  • LLM safety and observability -- keeping models (and the agents built on them) honest at inference time
  • The hardware-software boundary: SIMD, CUDA, kernel-level optimization
  • Novel architectures worth deploying, not just benchmarking

Things I care about less technically

  • Code that impresses interviewers but breaks on week two
  • Benchmarks that only win on synthetic data
  • Documentation that describes the happy path and nothing else

Selected work

Most of these exist because I needed to solve something concrete. I'd rather have a few things that are real than many that just look good on a profile.

ProjectWhat it isWhere it actually stands
Composed-MoE-Enginewrite-up: part I · part IITraining runtime for large Mixture-of-Experts models at hyperscale. Fused Triton top-k router, composable 4D parallelism (DP+EP+TP+PP), strict forward-pass invariants, elastic fault tolerance, async two-tier checkpointing with automatic expert resharding, chaos testing.Validated on real hardware, not just simulated: 80.1× (T4) and 58.7× (A100) kernel speedup, 348 tests, three hardware-only bugs found and documented. Multi-GPU (8×A100/8×H100) validation in progress. Cite
KANX -- write-up →Kolmogorov-Arnold Networks library with PyTorch + TensorFlow backends, verified ONNX export, Docker + Kubernetes + Helm support, FastAPI serving.pip install kanx · 5,000+ downloadsDownloadsCite
guardrail-rs -- write-up →Zero-Python LLM security layer written in Rust. Reverse-proxy that blocks prompt injection, redacts PII, and enforces policy with sub-millisecond overhead. Fails open by design, hot-reloadable config, Docker/K8s ready.Active. Six real bugs found and written up honestly -- see the write-up. Complements GuardRail Studio.
FlashSpec -- write-up →Adaptive speculative decoding engine with online bandit draft selection and Triton-optimised on-device verification. Preserves target distribution exactly.pip install flashspecDownloadsCite
mcp-reliableRuntime observability and reliability for the MCP ecosystem -- watches MCP servers while they're running: health checks, a traffic-observing proxy, schema/output drift detection, webhook + Slack alerting, and an agent-facing MCP interface so an agent can query its own tool reliability directly. CLI + REST API + dashboard, SQLite storage.v1.0.0 · 195 tests, none mocked at the protocol level · architecture fully documented via ADRs.
ReliableAgentReliability-first orchestration framework for agentic systems. Plan → execute → critique → replan loop with guardrails at every boundary, full trajectory reconstruction, checkpoint/resume, quantitative reliability metrics.v0.2.0 (June 19, 2026) · 140 tests
production-vlm-engineeringReproducible pipelines for modern multimodal vision systems: efficient VLM adaptation, embedding-space drift detection, edge inference, robustness & safety.Active.
RAG-Multimodal-Financial-Doc-Analysis-and-RecallMultimodal RAG system for financial documents -- hybrid retrieval, VLM chart understanding, numeric grounding, full observability + CI-gated evaluation.Active, finance-domain focused.
PromptCanary -- write-up →Detects silent LLM drift. Open Notebooks in Colabv0.2.2

Also digging into

Not everything is a shipped tool -- some of it is just a question I wanted a real answer to.

  • Does fine-tuning break a transformer's copy mechanism?Write-up →

Selected Open Source Contributions

These are small but high-signal fixes in core infrastructure projects:

ProjectStatusWhat was fixed / improvedLink
TritonMergedFixed NaN handling in tl.argmin/tl.argmax so interpreter matches JIT behaviorPR #10699
Megatron-LMMergedFixed crash in get_grad_norm_fp32 when gradient list is empty (common with frozen layers / tensor parallelism)PR #5530
TensorFlowMergedtf.experimental.numpy.swapaxes now raises a clear error on out-of-bounds axis instead of silent normalization or opaque XLA errorsPR #122544

Stack

Not a comprehensive list. Just what I actually reach for.

Training & inferencePyTorchTensorFlowTritonONNXTensorRTFSDP2TorchElastic

LLM ecosystemTransformersPEFT / LoRAvLLMLangChainFastAPITriton Inference Server

Distributed & infraNCCLKubernetesHelmTerraformAirflowRay

ObservabilityPrometheusGrafanaOpenTelemetryWeights & Biases

Low-levelC++AVX2 / SIMDCUDApybind11

DataPostgreSQLQdrantMongoDBSparkDask


Problem-solving

Algorithms are how I warm up. Systems are where I live.


A few honest notes

Most of my interesting work happens in private repositories -- production systems at cloud scale where open-sourcing isn't an option. This GitHub is a public window, not the full picture.

That said: the repositories here are written to the same standard I use privately: tests, type checking, CI, real (if limited) benchmarks, and documentation that tries to admit what doesn't work yet. When something is experimental or incomplete, the README says so.

I'm especially interested in the kinds of failures that only appear at real cluster scale, the practical trade-offs in LLM safety systems, and whether architectures like KANs will eventually find meaningful production use cases.

My path into this wasn't linear -- it started in mechatronics, building things with real sensors, actuators, and control loops that fail in ways no unit test catches, and later working on ML systems within offshore energy's integrated control and safety systems, where a silent failure has real consequences. That's the actual root of the reliability focus you see in these projects.


Currently

  • Working on: the MoE engine's chaos scenario A -- sudden node failure under expert resharding, currently recovering ~85% of the time
  • Reading: the Megatron-LM codebase and the FlexAttention paper
  • Thinking about: whether MFU tracking gives you enough signal to catch silent training degradation early

🎶 Current frequency

Spotify now playing

Outside of work I'm usually reading something I don't fully understand yet, listening to music that has no business being that good, and occasionally wondering if the model actually converged or if I just got lucky. I like working with people who say "I don't know" without embarrassment and argue about architecture in good faith.

Pinned Loading

  1. KANXKANXPublic

    One library, four surfaces. Production-grade Kolmogorov-Arnold Networks || TensorFlow + PyTorch + ONNX. || A small KAN beats a 10× larger MLP on smooth, separable target. One library. Two backends.…

    Python 33 8

  2. production-vlm-engineeringproduction-vlm-engineeringPublic

    Reproducible, production-grade pipelines for modern multimodal vision systems. Efficient VLM adaptation · Embedding-space drift detection · Edge inference · Robustness & safety

    Python 51

  3. Composed-Mixture-of-Experts-EngineComposed-Mixture-of-Experts-EnginePublic

    A research-grade fault-tolerant runtime for hyperscale Mixture-of-Experts training

    Python 12 8

  4. FlashSpecFlashSpecPublic

    Adaptive speculative-decoding inference engine with Triton-optimised verification and online bandit draft selection.

    Python 12

  5. guardrail-rsguardrail-rsPublic

    A zero-Python, production-grade LLM security layer written in Rust.

    Rust 11

  6. RAG-Multimodal-Financial-Doc-Analysis-and-RecallRAG-Multimodal-Financial-Doc-Analysis-and-RecallPublic

    Production-grade multimodal RAG for financial document intelligence. Chart understanding · hybrid retrieval · numeric guardrails · multi-tenancy · full observability.

    Python 75 18