tensorstudio 0.0.1a0__tar.gz → 0.1.0__tar.gz

tensorstudio-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,31 @@
+name: CI
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  test:
+    name: ${{ matrix.os }} py${{ matrix.python-version }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install
+        run: |
+          python -m pip install -U pip
+          python -m pip install -e ".[dev]"
+      - name: Ruff
+        run: ruff check .
+      - name: Mypy
+        run: mypy python/tensorstudio
+      - name: Tests
+        run: pytest

tensorstudio-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,39 @@
+name: Publish
+
+on:
+  release:
+    types: [published]
+  workflow_dispatch:
+
+permissions:
+  contents: read
+  id-token: write
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build distributions
+        run: |
+          python -m pip install -U build
+          python -m build
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/*
+
+  publish:
+    needs: build
+    runs-on: ubuntu-latest
+    environment: pypi
+    steps:
+      - uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

tensorstudio-0.1.0/.github/workflows/wheels.yml

@@ -0,0 +1,45 @@
+name: Wheels
+
+on:
+  push:
+    tags:
+      - "v*"
+  workflow_dispatch:
+
+jobs:
+  build_wheels:
+    name: Build wheels on ${{ matrix.os }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build wheels
+        uses: pypa/cibuildwheel@v2.23.3
+      - uses: actions/upload-artifact@v4
+        with:
+          name: wheels-${{ matrix.os }}
+          path: wheelhouse/*.whl
+
+  build_sdist:
+    name: Build sdist
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build sdist
+        run: |
+          python -m pip install -U build
+          python -m build --sdist
+      - uses: actions/upload-artifact@v4
+        with:
+          name: sdist
+          path: dist/*.tar.gz

tensorstudio-0.1.0/.gitignore

@@ -0,0 +1,21 @@
+build/
+dist/
+*.egg-info/
+__pycache__/
+*.py[cod]
+*.pyd
+*.so
+*.dylib
+.mypy_cache/
+.pytest_cache/
+.ruff_cache/
+.coverage
+htmlcov/
+site/
+.venv/
+venv/
+.env
+.env.*
+.idea/
+.vscode/
+*.tsmodel

tensorstudio-0.1.0/.pre-commit-config.yaml

@@ -0,0 +1,13 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.6.9
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.11.2
+    hooks:
+      - id: mypy
+        args: [python/tensorstudio]
+        additional_dependencies: [numpy>=1.24]

tensorstudio-0.1.0/CHANGELOG.md

@@ -0,0 +1,9 @@
+# Changelog
+
+## 0.1.0
+
+- Initial experimental release.
+- C++20 CPU tensor core with Python bindings.
+- Basic broadcasting, matrix multiplication, reductions, activations, and views.
+- Reverse-mode autograd for the initial operation set.
+- Python `nn`, `optim`, serialization, examples, tests, docs, and CI workflows.

tensorstudio-0.1.0/CMakeLists.txt

@@ -0,0 +1,51 @@
+cmake_minimum_required(VERSION 3.18)
+
+project(tensorstudio VERSION 0.1.0 LANGUAGES CXX)
+
+find_package(pybind11 CONFIG REQUIRED)
+
+set(TENSORSTUDIO_CORE_SOURCES
+  src/core/tensor.cpp
+  src/core/dtype.cpp
+  src/core/shape.cpp
+  src/core/device.cpp
+  src/core/storage.cpp
+  src/core/ops.cpp
+  src/core/autograd.cpp
+  src/core/module.cpp
+  src/core/optim.cpp
+  src/core/random.cpp
+  src/core/serialization.cpp
+  src/core/errors.cpp
+)
+
+set(TENSORSTUDIO_BINDING_SOURCES
+  src/bindings/bindings.cpp
+  src/bindings/bind_tensor.cpp
+  src/bindings/bind_ops.cpp
+  src/bindings/bind_autograd.cpp
+  src/bindings/bind_nn.cpp
+  src/bindings/bind_optim.cpp
+)
+
+pybind11_add_module(_C MODULE
+  ${TENSORSTUDIO_CORE_SOURCES}
+  ${TENSORSTUDIO_BINDING_SOURCES}
+)
+
+target_include_directories(_C PRIVATE include src/bindings)
+target_compile_features(_C PRIVATE cxx_std_20)
+
+if(MSVC)
+  target_compile_options(_C PRIVATE /W4 /permissive-)
+else()
+  target_compile_options(_C PRIVATE -Wall -Wextra -Wpedantic)
+endif()
+
+target_compile_definitions(_C PRIVATE TENSORSTUDIO_VERSION="${PROJECT_VERSION}")
+
+install(TARGETS _C
+  LIBRARY DESTINATION tensorstudio
+  RUNTIME DESTINATION tensorstudio
+  ARCHIVE DESTINATION tensorstudio
+)

tensorstudio-0.1.0/CONTRIBUTING.md

@@ -0,0 +1,30 @@
+# Contributing
+
+Thank you for helping improve TensorStudio.
+
+## Development Setup
+
+```bash
+python -m pip install -U pip
+python -m pip install -e ".[dev]"
+pre-commit install
+```
+
+## Checks
+
+Run these before opening a pull request:
+
+```bash
+ruff check .
+mypy python/tensorstudio
+pytest
+python -m build
+```
+
+## Guidelines
+
+- Keep C++ tensor operations clear and maintainable.
+- Add tests for user-visible behavior changes.
+- Document limitations honestly.
+- Do not introduce network calls in library code.
+- Do not use `eval` or `exec`.

tensorstudio-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 TensorStudio contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

tensorstudio-0.1.0/PKG-INFO

@@ -0,0 +1,188 @@
+Metadata-Version: 2.2
+Name: tensorstudio
+Version: 0.1.0
+Summary: TensorStudio is a compact C++ tensor and autograd engine with a Python API for learning, experimentation, and lightweight ML workloads.
+Keywords: tensor,autograd,machine-learning,cpp,pybind11
+Author: TensorStudio contributors
+License: MIT License
+         
+         Copyright (c) 2026 TensorStudio contributors
+         
+         Permission is hereby granted, free of charge, to any person obtaining a copy
+         of this software and associated documentation files (the "Software"), to deal
+         in the Software without restriction, including without limitation the rights
+         to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+         copies of the Software, and to permit persons to whom the Software is
+         furnished to do so, subject to the following conditions:
+         
+         The above copyright notice and this permission notice shall be included in all
+         copies or substantial portions of the Software.
+         
+         THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+         IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+         FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+         AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+         LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+         OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+         SOFTWARE.
+         
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: C++
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Project-URL: Homepage, https://github.com/tensorstudio/tensorstudio
+Project-URL: Repository, https://github.com/tensorstudio/tensorstudio
+Project-URL: Issues, https://github.com/tensorstudio/tensorstudio/issues
+Requires-Python: >=3.10
+Requires-Dist: numpy>=1.24
+Provides-Extra: dev
+Requires-Dist: pytest>=8; extra == "dev"
+Requires-Dist: pytest-cov>=5; extra == "dev"
+Requires-Dist: build>=1.2; extra == "dev"
+Requires-Dist: twine>=5; extra == "dev"
+Requires-Dist: ruff>=0.6; extra == "dev"
+Requires-Dist: mypy>=1.10; extra == "dev"
+Requires-Dist: cibuildwheel>=2.19; extra == "dev"
+Requires-Dist: pre-commit>=3.7; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: mkdocs>=1.6; extra == "docs"
+Requires-Dist: mkdocs-material>=9.5; extra == "docs"
+Description-Content-Type: text/markdown
+
+# TensorStudio
+
+[![CI](https://github.com/tensorstudio/tensorstudio/actions/workflows/ci.yml/badge.svg)](https://github.com/tensorstudio/tensorstudio/actions/workflows/ci.yml)
+[![Wheels](https://github.com/tensorstudio/tensorstudio/actions/workflows/wheels.yml/badge.svg)](https://github.com/tensorstudio/tensorstudio/actions/workflows/wheels.yml)
+[![PyPI](https://img.shields.io/pypi/v/tensorstudio.svg)](https://pypi.org/project/tensorstudio/)
+
+TensorStudio is a compact C++ tensor and autograd engine with a Python API for
+learning, experimentation, and lightweight ML workloads.
+
+TensorStudio is experimental v0.1.0 software. It is CPU-only, eager-only, and
+intentionally small enough to read and modify.
+
+## Install
+
+From a source checkout:
+
+```bash
+python -m pip install -U pip
+python -m pip install -e ".[dev]"
+```
+
+Build source and wheel distributions:
+
+```bash
+python -m build
+```
+
+## Quickstart
+
+```python
+import tensorstudio as ts
+
+x = ts.tensor([[1.0, 2.0], [3.0, 4.0]])
+y = ts.ones((2, 2))
+
+print((x + y).tolist())
+print((x @ y).numpy())
+```
+
+## Autograd
+
+```python
+import tensorstudio as ts
+
+x = ts.tensor([1.0, 2.0, 3.0], requires_grad=True)
+loss = ((x * x).sum())
+loss.backward()
+
+print(x.grad.tolist())  # [2.0, 4.0, 6.0]
+```
+
+## Neural Networks
+
+```python
+import tensorstudio as ts
+from tensorstudio import nn, optim
+
+model = nn.Sequential(nn.Linear(1, 8), nn.Tanh(), nn.Linear(8, 1))
+optimizer = optim.SGD(model.parameters(), lr=0.05)
+criterion = nn.MSELoss()
+
+x = ts.tensor([[0.0], [1.0], [2.0], [3.0]])
+y = ts.tensor([[1.0], [3.0], [5.0], [7.0]])
+
+for _ in range(50):
+    optimizer.zero_grad()
+    loss = criterion(model(x), y)
+    loss.backward()
+    optimizer.step()
+
+print(loss.item())
+```
+
+## Development
+
+```bash
+python -m pip install -e ".[dev]"
+pytest
+ruff check .
+mypy python/tensorstudio
+python -m build
+```
+
+The native extension is built with CMake, pybind11, scikit-build-core, and C++20.
+
+## Documentation
+
+The documentation lives in `docs/` and is configured with MkDocs:
+
+```bash
+python -m pip install -e ".[docs]"
+mkdocs serve
+mkdocs build
+```
+
+The docs cover tensor semantics, broadcasting, NumPy interop, autograd, neural
+network modules, training loops, CPU backend details, benchmarks, development,
+publishing, and the project roadmap.
+
+## Publishing
+
+Releases are intended to be built by GitHub Actions. The publish workflow uses
+PyPI trusted publishing through `pypa/gh-action-pypi-publish`; do not commit PyPI
+API tokens.
+
+## Current Limitations
+
+- CPU backend only
+- Eager execution only
+- No CUDA kernels or mixed precision
+- No graph compiler or distributed training
+- Limited dtype casting and no advanced indexing
+- No sparse tensors
+- Pickle serialization is for trusted TensorStudio objects only
+
+## Roadmap
+
+- CUDA backend
+- Graph/JIT mode
+- Convolution ops
+- Dataset utilities
+- Model zoo examples
+- ONNX import/export
+- Improved memory allocator
+- SIMD kernels
+- Multithreaded ops
+
+## License
+
+TensorStudio is licensed under the MIT License.

tensorstudio-0.1.0/README.md

@@ -0,0 +1,130 @@
+# TensorStudio
+
+[![CI](https://github.com/tensorstudio/tensorstudio/actions/workflows/ci.yml/badge.svg)](https://github.com/tensorstudio/tensorstudio/actions/workflows/ci.yml)
+[![Wheels](https://github.com/tensorstudio/tensorstudio/actions/workflows/wheels.yml/badge.svg)](https://github.com/tensorstudio/tensorstudio/actions/workflows/wheels.yml)
+[![PyPI](https://img.shields.io/pypi/v/tensorstudio.svg)](https://pypi.org/project/tensorstudio/)
+
+TensorStudio is a compact C++ tensor and autograd engine with a Python API for
+learning, experimentation, and lightweight ML workloads.
+
+TensorStudio is experimental v0.1.0 software. It is CPU-only, eager-only, and
+intentionally small enough to read and modify.
+
+## Install
+
+From a source checkout:
+
+```bash
+python -m pip install -U pip
+python -m pip install -e ".[dev]"
+```
+
+Build source and wheel distributions:
+
+```bash
+python -m build
+```
+
+## Quickstart
+
+```python
+import tensorstudio as ts
+
+x = ts.tensor([[1.0, 2.0], [3.0, 4.0]])
+y = ts.ones((2, 2))
+
+print((x + y).tolist())
+print((x @ y).numpy())
+```
+
+## Autograd
+
+```python
+import tensorstudio as ts
+
+x = ts.tensor([1.0, 2.0, 3.0], requires_grad=True)
+loss = ((x * x).sum())
+loss.backward()
+
+print(x.grad.tolist())  # [2.0, 4.0, 6.0]
+```
+
+## Neural Networks
+
+```python
+import tensorstudio as ts
+from tensorstudio import nn, optim
+
+model = nn.Sequential(nn.Linear(1, 8), nn.Tanh(), nn.Linear(8, 1))
+optimizer = optim.SGD(model.parameters(), lr=0.05)
+criterion = nn.MSELoss()
+
+x = ts.tensor([[0.0], [1.0], [2.0], [3.0]])
+y = ts.tensor([[1.0], [3.0], [5.0], [7.0]])
+
+for _ in range(50):
+    optimizer.zero_grad()
+    loss = criterion(model(x), y)
+    loss.backward()
+    optimizer.step()
+
+print(loss.item())
+```
+
+## Development
+
+```bash
+python -m pip install -e ".[dev]"
+pytest
+ruff check .
+mypy python/tensorstudio
+python -m build
+```
+
+The native extension is built with CMake, pybind11, scikit-build-core, and C++20.
+
+## Documentation
+
+The documentation lives in `docs/` and is configured with MkDocs:
+
+```bash
+python -m pip install -e ".[docs]"
+mkdocs serve
+mkdocs build
+```
+
+The docs cover tensor semantics, broadcasting, NumPy interop, autograd, neural
+network modules, training loops, CPU backend details, benchmarks, development,
+publishing, and the project roadmap.
+
+## Publishing
+
+Releases are intended to be built by GitHub Actions. The publish workflow uses
+PyPI trusted publishing through `pypa/gh-action-pypi-publish`; do not commit PyPI
+API tokens.
+
+## Current Limitations
+
+- CPU backend only
+- Eager execution only
+- No CUDA kernels or mixed precision
+- No graph compiler or distributed training
+- Limited dtype casting and no advanced indexing
+- No sparse tensors
+- Pickle serialization is for trusted TensorStudio objects only
+
+## Roadmap
+
+- CUDA backend
+- Graph/JIT mode
+- Convolution ops
+- Dataset utilities
+- Model zoo examples
+- ONNX import/export
+- Improved memory allocator
+- SIMD kernels
+- Multithreaded ops
+
+## License
+
+TensorStudio is licensed under the MIT License.

tensorstudio-0.1.0/SECURITY.md

@@ -0,0 +1,21 @@
+# Security Policy
+
+TensorStudio is experimental v0.1.0 software.
+
+## Reporting a Vulnerability
+
+Please report suspected vulnerabilities by opening a private security advisory
+on GitHub or by contacting the maintainers through the project repository.
+
+Include:
+
+- Affected version or commit
+- Reproduction steps
+- Impact assessment
+- Suggested fix, if available
+
+## Serialization Warning
+
+`tensorstudio.load()` uses Python pickle for v0.1.0 internal object roundtrips.
+Loading pickle files from untrusted sources is unsafe because pickle can execute
+arbitrary code during deserialization.

tensorstudio-0.1.0/benchmarks/bench_matmul.py

@@ -0,0 +1,28 @@
+from __future__ import annotations
+
+import time
+
+import tensorstudio as ts
+
+
+def timeit(name: str, fn, iterations: int = 5) -> None:
+    start = time.perf_counter()
+    for _ in range(iterations):
+        fn()
+    elapsed = (time.perf_counter() - start) / iterations
+    print(f"{name}: {elapsed * 1_000:.3f} ms")
+
+
+def main() -> None:
+    shape = (128, 128)
+    a_ts = ts.randn(shape, seed=1)
+    b_ts = ts.randn(shape, seed=2)
+    a_np = a_ts.numpy()
+    b_np = b_ts.numpy()
+
+    timeit("tensorstudio matmul", lambda: a_ts @ b_ts)
+    timeit("numpy matmul", lambda: a_np @ b_np)
+
+
+if __name__ == "__main__":
+    main()

tensorstudio-0.1.0/benchmarks/bench_tensor_ops.py

@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+import time
+
+import numpy as np
+import tensorstudio as ts
+
+
+def timeit(name: str, fn, iterations: int = 20) -> None:
+    start = time.perf_counter()
+    for _ in range(iterations):
+        fn()
+    elapsed = (time.perf_counter() - start) / iterations
+    print(f"{name}: {elapsed * 1_000:.3f} ms")
+
+
+def main() -> None:
+    size = 50_000
+    a_ts = ts.randn((size,), seed=1)
+    b_ts = ts.randn((size,), seed=2)
+    a_np = a_ts.numpy()
+    b_np = b_ts.numpy()
+
+    timeit("tensorstudio add", lambda: a_ts + b_ts)
+    timeit("numpy add", lambda: a_np + b_np)
+    timeit("tensorstudio relu", lambda: a_ts.relu())
+    timeit("numpy relu", lambda: np.maximum(a_np, 0))
+
+
+if __name__ == "__main__":
+    main()