VibeTensor: System Software for Deep Learning, Fully Generated by AI Agents
On the map: LLM agents and tool use → Software engineering coding agents
Abstract (arXiv)
VIBETENSOR is an open-source research system software stack for deep learning, generated by LLM-powered coding agents under high-level human guidance. In this paper, "fully generated" refers to code provenance: implementation changes were produced and applied as agent-proposed diffs; validation relied on agent-run builds, tests, and differential checks, without per-change manual diff review. It implements a PyTorch-style eager tensor library with a C++20 core (CPU+CUDA), a torch-like Python overlay via nanobind, and an experimental Node.js/TypeScript interface. Unlike thin bindings, VIBETENSOR includes its own tensor/storage system, schema-lite dispatcher, reverse-mode autograd, CUDA runtime (streams/events/graphs), a stream-ordered caching allocator with diagnostics, and a stable C ABI for dynamically loaded operator plugins. We view this release as a milestone for AI-assisted software engineering: it shows coding agents can generate a coherent deep learning runtime spanning language bindings down to CUDA memory management, validated primarily by builds and tests. We describe the architecture, summarize the workflow used to produce and validate the system, and evaluate the artifact. We report repository scale and test-suite composition, and summarize reproducible microbenchmarks from an accompanying AI-generated kernel suite, including fused attention versus PyTorch SDPA/FlashAttention. We also report end-to-end training sanity checks on 3 small workloads (sequence reversal, ViT, miniGPT) on NVIDIA H100 (Hopper, SM90) and Blackwell-class GPUs; multi-GPU results are Blackwell-only and use an optional CUTLASS-based ring-allreduce plugin gated on CUDA 13+ and sm103a toolchain support. Finally, we discuss failure modes in generated system software, including a "Frankenstein" composition effect where locally correct subsystems interact to yield globally suboptimal performance.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Software engineering coding agents, one of 16 directions in LLM agents and tool use. Builds on 7 of its 15 references that are in the map. First posted on arXiv 21 Jan 2026.
Builds on
7 of 15 references.
References · 7
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,373
- 2021Escaping the Big Data Paradigm with Compact Transformerscited by 264
- 2017DyNet: The Dynamic Neural Network Toolkitcited by 21
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Searchcited by 24
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limitscited by 18
- 2026FM-Agent: Scaling Formal Methods to Large Systems via LLM-Based Hoare-Style Reasoningcited by 6
- 2026Computer Architecture's AlphaZero Moment: Automated Discovery in an Encircled Worldcited by 5
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPUcited by 2
- 2026stratum: A System Infrastructure for Massive Agent-Centric ML Workloadscited by 1
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPU16 Apr 2026
- 2026FM-Agent: Scaling Formal Methods to Large Systems via LLM-Based Hoare-Style Reasoning13 Apr 2026
- 2026Computer Architecture's AlphaZero Moment: Automated Discovery in an Encircled World31 Mar 2026
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Search25 Mar 2026
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limits19 Mar 2026
- 2026stratum: A System Infrastructure for Massive Agent-Centric ML Workloads3 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2026ATTest: Agent-Driven Tensor Testing for Deep Learning Library Modules
- 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels
- 2026TensorBench: Benchmarking Coding Agents on a Compiler-Based Tensor Framework
- 2026CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
- 2025Vibe Coding: Toward an AI-Native Paradigm for Semantic and Intent-Driven Programming ↗
- 2026Agentic Framework for Deep Learning workload migration via In-Context Learning
- 2026Tensor Algebraic Property Skeletons: Amplifying Property-Based Testing for AI Compilers ↗
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphs