Skip to main content

High-performance zero-copy tensor protocol

Project description

Gemini_Generated_Image_v39t46v39t46v39t

Tenso

High-Performance, Zero-Copy Tensor Protocol for Python.

Overview

Tenso is a specialized binary protocol designed for one thing: moving NumPy arrays between backends instantly.

It avoids the massive CPU overhead of standard formats (JSON, Pickle, MsgPack) by using a strict Little-Endian, 64-byte aligned memory layout. This allows for Zero-Copy deserialization, meaning the CPU doesn't have to move data—it just points to it.

The Zero-CPU Advantage

Tenso isn't just about speed; it's about resource efficiency.

  • JSON/Pickle: Parsing large arrays consumes significant CPU cycles (100% usage during load). In a high-throughput cluster, this steals resources from your actual model inference.

  • Tenso: Deserialization is effectively 0% CPU. The processor simply maps the existing memory address.

Benchmark

Scenario: Reading a 64MB Float32 Matrix (Typical LLM Layer) from memory.

Format Read Time Write Time Status
Tenso 0.006 ms 5.287 ms Fastest & AVX-512 Safe
Arrow 0.007 ms 7.368 ms Heavy Dependency
Pickle 2.670 ms 2.773 ms Unsafe (Security Risk)
Safetensors 2.489 ms 7.747 ms -
MsgPack 2.536 ms 10.830 ms -

Installation

pip install tenso

Usage

Network

import numpy as np
import tenso

# Create a tensor
data = np.random.rand(100, 100).astype(np.float32)

# Serialize to bytes
packet = tenso.dumps(data)

# Deserialize back
restored = tenso.loads(packet)

File I/O

# Load from disk (Standard)
with open("weights.tenso", "rb") as f:
    loaded_data = tenso.load(f)

# Load Large Models (Larger than RAM)
# Uses OS memory mapping to read data instantly without loading file into memory
with open("llama_70b_weights.tenso", "rb") as f:
    loaded_data = tenso.load(f, mmap_mode=True)

Protocol Specification

Tenso uses a Hybrid Fixed-Header format designed for SIMD safety.

  • Header (8 bytes): TNSO Magic, Version, Flags, Dtype, NDim.

  • Shape Block: Variable length (NDim * 4 bytes).

  • Padding: 0-63 bytes to ensure the Body starts at a 64-byte aligned address.

  • Body: Raw C-contiguous memory dump.

Tenso vs. The World

Feature Tenso Pickle Arrow Safetensors
Speed (Read) Instant Slow Instant Fast
Safety Secure Unsafe (RCE Risk) Secure Secure
Alignment 64-byte None 64-byte None
Dependencies NumPy Only Python PyArrow (Huge) Rust/Bindings
Best For Network/IPC Python Objects Dataframes Disk Storage

Why is Tenso 1500x faster than Pickle? Standard formats must copy data from the network buffer into a new NumPy array. Tenso uses Memory Mapping: it tells NumPy to point directly at the existing buffer. No copying, no CPU cycles.

Development

# Clone the repository
git clone https://github.com/yourusername/tenso.git
cd tenso

# Install development dependencies
pip install -e ".[dev]"

# Run tests
pytest

# Run benchmarks
python benchmark.py

Advanced Usage

Strict Mode: Prevent accidental memory copies during serialization. Raises an error if data is not already C-Contiguous.

try:
    # Will raise ValueError if array is Fortran-contiguous or non-contiguous
    packet = tenso.dumps(data, strict=True) 
except ValueError:
    print("Array must be C-Contiguous!")

Requirements

  • Python >= 3.10
  • NumPy

License

This project is licensed under the MIT License - see the LICENSE file for details.

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tenso-0.3.2.tar.gz (4.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tenso-0.3.2-py3-none-any.whl (6.2 kB view details)

Uploaded Python 3

File details

Details for the file tenso-0.3.2.tar.gz.

File metadata

  • Download URL: tenso-0.3.2.tar.gz
  • Upload date:
  • Size: 4.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.9

File hashes

Hashes for tenso-0.3.2.tar.gz
Algorithm Hash digest
SHA256 9943ddd45b2250fbe7527097808ebed96a5ebb28df4dfcb9649570764bdad107
MD5 463c409dde5d108866f916dd3ebb2215
BLAKE2b-256 3f2b6c5d4b6ca5b1299878637ec0bc4dda306a5370b1b79b2c4a6df995766935

See more details on using hashes here.

File details

Details for the file tenso-0.3.2-py3-none-any.whl.

File metadata

  • Download URL: tenso-0.3.2-py3-none-any.whl
  • Upload date:
  • Size: 6.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.9

File hashes

Hashes for tenso-0.3.2-py3-none-any.whl
Algorithm Hash digest
SHA256 342dcdcaa1b1e23af088d7a835ed16ae1f4d60ad9a2b145057ffdf39d868fe2a
MD5 f47888a1a96dd89c59bf6a03b4e3b123
BLAKE2b-256 92a978d00df9372bcdd63d04f14bc1563b456979bcea2d952f60736436e78fd0

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page