Skip to main content

Production neuromorphic inference SDK with CUDA acceleration

Project description

NeuraTensor SDK

Production neuromorphic inference at 8ms latency

NeuraTensor is a high-performance neuromorphic inference runtime featuring a proprietary fused SNN-SSM architecture with CUDA acceleration. Optimized for NVIDIA Jetson AGX Orin and edge deployment.

PyPI version Python 3.8+ CUDA 11.4+


⚡ Performance

  • 8.13ms mean latency (Jetson AGX Orin, batch=1, seq=64, FP16)
  • 123 sequences/sec throughput
  • Sub-10ms real-time inference guarantee
  • Up to 120× lower latency vs standard PyTorch execution (measured on Jetson AGX Orin, batch=1, seq=64, same model architecture)

🚀 Quick Start

Installation

pip install neuratensor

Python API

from neuratensor import NeuraTensor, NeuraTensorConfig
import torch

# Load model (3 lines)
config = NeuraTensorConfig.preset("64m")
model = NeuraTensor(config).cuda().half()

# Inference (1 line)
input_ids = torch.randint(0, 50000, (1, 64), device="cuda")
output = model(input_ids)
# ✅ 8ms latency, 64M params, 123 seq/s

CLI Benchmark

# Run benchmark
neuratensor benchmark 64m --iterations 100

# Example output:
# ============================================================
# NeuraTensor 64M | Benchmark
# ============================================================
# Device:    Orin (SM 8.7)
# CUDA:      11.4
# ------------------------------------------------------------
# Latency (mean):    8.13 ms ± 0.07 ms
# Latency (p99):     8.22 ms
# Throughput:        123.0 seq/s
# Kernel:            fused_snn_ssm (secure)
# ============================================================

🏗️ Architecture

  • Hybrid SNN-SSM: 6 spiking layers + 10 state space layers
  • Fused CUDA kernels: Single-pass execution (patent pending)
  • FP16 optimized: Hardware tensor cores
  • Deterministic latency: No dynamic dispatch overhead

📦 Available Models

Model Parameters Latency (Orin) Throughput Use Case
64M 64,155,526 8.13ms 123 seq/s Edge, real-time
256M 256M ~15ms 67 seq/s Balanced
1B 1B ~35ms 29 seq/s Quality

All measurements: Jetson AGX Orin, batch=1, seq=64, FP16


⚠️ What This Is NOT

NeuraTensor is NOT:

  • ❌ A training framework
  • ❌ A general-purpose Transformer library
  • ❌ A drop-in replacement for PyTorch
  • ❌ An autograd-enabled tensor library

NeuraTensor IS:

  • ✅ A production-grade inference runtime
  • ✅ Optimized for edge latency and determinism
  • ✅ Purpose-built for neuromorphic architectures
  • ✅ A compiled binary SDK with Python bindings

💻 Supported Hardware

Tested & Validated:

  • Jetson AGX Orin (primary target)
  • Jetson Orin NX
  • Jetson Orin Nano (limited testing)

Experimental Support:

  • ⚠️ NVIDIA RTX 30xx / 40xx (desktop GPUs)
  • ⚠️ A100, H100 (datacenter GPUs)

Requirements:

  • CUDA Compute Capability: 7.0 or higher (SM 7.0+)
  • CUDA Toolkit: 11.4 or higher
  • Driver: 470.x or higher
  • Python: 3.8, 3.9, 3.10
  • PyTorch: 2.0+ (for tensor compatibility only)

Note: If your GPU is unsupported, the SDK will raise a clear error at import time. Source code for the core runtime is not distributed.


📦 Binary Distribution Model

The SDK ships with precompiled CUDA binaries for maximum performance:

  • ✅ Optimized for Jetson AGX Orin (ARM64 + CUDA 11.4)
  • ✅ No compilation required at install time
  • ✅ Symbols obfuscated for IP protection
  • ⚠️ Platform-specific (Linux ARM64 only for v1.x)

If your system configuration is unsupported, you will see:

RuntimeError: CUDA kernel not compatible with this device

Source code for the proprietary core is not distributed. Binary-only distribution is intentional.


🔒 Security & IP Notice

NeuraTensor contains proprietary CUDA kernels and runtime logic protected by:

  • Patent-pending architecture (USPTO filing: pending)
  • Obfuscated binaries (symbols stripped and renamed)
  • Restricted license (see PATENT_NOTICE.txt)

Reverse engineering, redistribution, or modification of the core binaries is prohibited by the license.

For commercial licensing, custom hardware support, or source code access:


🔧 Requirements

System:

  • OS: Linux (Ubuntu 20.04+ recommended)
  • Architecture: ARM64 (Jetson) or x86_64 (experimental)
  • CUDA: 11.4+ with cuDNN
  • GPU Memory: 2GB minimum

Python:

pip install neuratensor

# Dependencies (auto-installed):
# - torch >= 2.0.0
# - numpy >= 1.19.0

📚 Documentation

  • Quick Start: examples/
  • API Reference: Coming soon
  • Performance Guide: Coming soon
  • Hardware Guide: Coming soon

🤝 Support

Community:

Enterprise:


📊 Citation

If you use NeuraTensor in your research or product, please cite:

@software{neuratensor2025,
  title = {NeuraTensor: High-Performance Neuromorphic Inference SDK},
  author = {Neuramorphic, Inc.},
  year = {2025},
  url = {https://pypi.org/project/neuratensor/}
}

📄 License

Proprietary License - Binary distribution only.

  • ✅ Evaluation and non-commercial use permitted
  • ⚠️ Commercial use requires separate license agreement
  • ❌ Reverse engineering strictly prohibited
  • ❌ Redistribution not permitted

See LICENSE and PATENT_NOTICE.txt for details.


🏢 About

NeuraTensor is developed by Neuramorphic, Inc.

Patent-pending neuromorphic architecture combining spiking neural networks with state space models for ultra-low latency inference on edge devices.


© 2024-2025 Neuramorphic, Inc. All rights reserved.

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

neuratensor-1.1.4.tar.gz (253.8 kB view details)

Uploaded Source

Built Distribution

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

neuratensor-1.1.4-py3-none-any.whl (45.1 kB view details)

Uploaded Python 3

File details

Details for the file neuratensor-1.1.4.tar.gz.

File metadata

  • Download URL: neuratensor-1.1.4.tar.gz
  • Upload date:
  • Size: 253.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.8.10

File hashes

Hashes for neuratensor-1.1.4.tar.gz
Algorithm Hash digest
SHA256 380630245908a6f642649aa24dc797f088dad2373590d19c0f662c047cf82383
MD5 ac0955335a7cc49bd150f6db3dde9d2b
BLAKE2b-256 942418be2ca6eacd7a71335f6d4251c47cdef0ede1bc5ab7c53aadc057da6487

See more details on using hashes here.

File details

Details for the file neuratensor-1.1.4-py3-none-any.whl.

File metadata

  • Download URL: neuratensor-1.1.4-py3-none-any.whl
  • Upload date:
  • Size: 45.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.8.10

File hashes

Hashes for neuratensor-1.1.4-py3-none-any.whl
Algorithm Hash digest
SHA256 a59a214f38a3c5179ab8f5db346898c3ecf6002a9c30d5a29abeaf2a42d8dc74
MD5 4f0a85d769e6057c0d5103ec2edfc519
BLAKE2b-256 522e59787e1c306e9bb3d8e6cc69f42f114112df452bc6cffd3d61734b6992e2

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