tensorstudio 0.1.1__tar.gz → 1.0.0__tar.gz

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

@@ -0,0 +1,92 @@
+name: CI
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  windows:
+    name: Windows py${{ matrix.python-version }}
+    runs-on: windows-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        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 editable package
+        run: |
+          python -m pip install -U pip
+          python -m pip install -e ".[dev]"
+      - name: Import native extension
+        run: python -c "import tensorstudio as ts; import tensorstudio._C; print(ts.__version__); print(ts.ones((2, 2)) + 1)"
+      - name: Ruff
+        run: ruff check .
+      - name: Mypy
+        run: mypy --python-version ${{ matrix.python-version }} python/tensorstudio
+      - name: Tests
+        run: pytest -q
+      - name: Example
+        run: python examples/basic_tensor_ops.py
+
+  linux:
+    name: Linux py${{ matrix.python-version }}
+    runs-on: ubuntu-latest
+    needs: windows
+    strategy:
+      fail-fast: false
+      matrix:
+        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 editable package
+        run: |
+          python -m pip install -U pip
+          python -m pip install -e ".[dev]"
+      - name: Import native extension
+        run: python -c "import tensorstudio as ts; import tensorstudio._C; print(ts.__version__)"
+      - name: Ruff
+        run: ruff check .
+      - name: Mypy
+        run: mypy --python-version ${{ matrix.python-version }} python/tensorstudio
+      - name: Tests
+        run: pytest -q
+      - name: Example
+        run: python examples/basic_tensor_ops.py
+
+  macos:
+    name: macOS py${{ matrix.python-version }}
+    runs-on: macos-latest
+    needs: linux
+    strategy:
+      fail-fast: false
+      matrix:
+        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 editable package
+        run: |
+          python -m pip install -U pip
+          python -m pip install -e ".[dev]"
+      - name: Import native extension
+        run: python -c "import tensorstudio as ts; import tensorstudio._C; print(ts.__version__)"
+      - name: Ruff
+        run: ruff check .
+      - name: Mypy
+        run: mypy --python-version ${{ matrix.python-version }} python/tensorstudio
+      - name: Tests
+        run: pytest -q
+      - name: Example
+        run: python examples/basic_tensor_ops.py

tensorstudio-1.0.0/.github/workflows/publish-testpypi.yml

@@ -0,0 +1,108 @@
+name: Publish TestPyPI
+
+on:
+  workflow_dispatch:
+
+permissions:
+  contents: read
+  id-token: write
+
+env:
+  CIBW_BUILD: "cp310-* cp311-* cp312-* cp313-*"
+  CIBW_SKIP: "*-musllinux*"
+  CIBW_MANYLINUX_X86_64_IMAGE: "manylinux_2_28"
+  CIBW_TEST_COMMAND: >-
+    python -c "import tensorstudio as ts; import tensorstudio._C; x = ts.ones((2, 2), requires_grad=True); y = ((x * 2.0) + 1.0).sum(); y.backward(); assert x.grad.tolist() == [[2.0, 2.0], [2.0, 2.0]]"
+
+jobs:
+  windows_wheels:
+    name: Windows wheels
+    runs-on: windows-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build Windows wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_WINDOWS: "AMD64"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-windows
+          path: wheelhouse/*.whl
+
+  linux_wheels:
+    name: Linux wheels
+    runs-on: ubuntu-latest
+    needs: windows_wheels
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build Linux wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_LINUX: "x86_64"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-linux
+          path: wheelhouse/*.whl
+
+  macos_wheels:
+    name: macOS wheels
+    runs-on: macos-latest
+    needs: linux_wheels
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build macOS wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_MACOS: "auto"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-macos
+          path: wheelhouse/*.whl
+
+  build_sdist:
+    name: Build sdist
+    runs-on: ubuntu-latest
+    needs: macos_wheels
+    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 twine
+          python -m build --sdist
+          python -m twine check dist/*
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-sdist
+          path: dist/*.tar.gz
+
+  publish:
+    name: Publish to TestPyPI
+    needs: [windows_wheels, linux_wheels, macos_wheels, build_sdist]
+    runs-on: ubuntu-latest
+    environment: testpypi
+    steps:
+      - uses: actions/download-artifact@v4
+        with:
+          pattern: dist-*
+          path: dist
+          merge-multiple: true
+      - name: Verify artifacts
+        run: |
+          python -m pip install -U twine
+          python -m twine check dist/*
+      - name: Publish to TestPyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/

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

@@ -0,0 +1,107 @@
+name: Publish
+
+on:
+  workflow_dispatch:
+  release:
+    types: [published]
+  push:
+    tags:
+      - "v1.0.0"
+
+permissions:
+  contents: read
+  id-token: write
+
+env:
+  CIBW_BUILD: "cp310-* cp311-* cp312-* cp313-*"
+  CIBW_SKIP: "*-musllinux*"
+  CIBW_MANYLINUX_X86_64_IMAGE: "manylinux_2_28"
+  CIBW_TEST_COMMAND: >-
+    python -c "import tensorstudio as ts; import tensorstudio._C; x = ts.ones((2, 2), requires_grad=True); y = ((x * 2.0) + 1.0).sum(); y.backward(); assert x.grad.tolist() == [[2.0, 2.0], [2.0, 2.0]]; print(ts.__version__, x.grad.tolist())"
+
+jobs:
+  windows_wheels:
+    name: Windows wheels
+    runs-on: windows-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build Windows wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_WINDOWS: "AMD64"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-windows
+          path: wheelhouse/*.whl
+
+  linux_wheels:
+    name: Linux wheels
+    runs-on: ubuntu-latest
+    needs: windows_wheels
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build Linux wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_LINUX: "x86_64"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-linux
+          path: wheelhouse/*.whl
+
+  macos_wheels:
+    name: macOS wheels
+    runs-on: macos-latest
+    needs: linux_wheels
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build macOS wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_MACOS: "auto"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-macos
+          path: wheelhouse/*.whl
+
+  build_sdist:
+    name: Build sdist
+    runs-on: ubuntu-latest
+    needs: macos_wheels
+    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 twine
+          python -m build --sdist
+          python -m twine check dist/*
+      - uses: actions/upload-artifact@v4
+        with:
+          name: dist-sdist
+          path: dist/*.tar.gz
+
+  publish:
+    name: Publish to PyPI
+    needs: [windows_wheels, linux_wheels, macos_wheels, build_sdist]
+    runs-on: ubuntu-latest
+    environment: pypi
+    steps:
+      - uses: actions/download-artifact@v4
+        with:
+          pattern: dist-*
+          path: dist
+          merge-multiple: true
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

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

@@ -0,0 +1,90 @@
+name: Wheels
+
+on:
+  push:
+    tags:
+      - "v*"
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+env:
+  CIBW_BUILD: "cp310-* cp311-* cp312-* cp313-*"
+  CIBW_SKIP: "*-musllinux*"
+  CIBW_MANYLINUX_X86_64_IMAGE: "manylinux_2_28"
+  CIBW_TEST_COMMAND: >-
+    python -c "import tensorstudio as ts; import tensorstudio._C; x = ts.ones((2, 2), requires_grad=True); y = ((x * 2.0) + 1.0).sum(); y.backward(); assert x.grad.tolist() == [[2.0, 2.0], [2.0, 2.0]]; print(ts.__version__, x.grad.tolist())"
+
+jobs:
+  windows_wheels:
+    name: Windows wheels
+    runs-on: windows-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build Windows wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_WINDOWS: "AMD64"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: wheels-windows
+          path: wheelhouse/*.whl
+
+  linux_wheels:
+    name: Linux wheels
+    runs-on: ubuntu-latest
+    needs: windows_wheels
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build Linux wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_LINUX: "x86_64"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: wheels-linux
+          path: wheelhouse/*.whl
+
+  macos_wheels:
+    name: macOS wheels
+    runs-on: macos-latest
+    needs: linux_wheels
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Build macOS wheels
+        uses: pypa/cibuildwheel@v2.23.3
+        env:
+          CIBW_ARCHS_MACOS: "auto"
+      - uses: actions/upload-artifact@v4
+        with:
+          name: wheels-macos
+          path: wheelhouse/*.whl
+
+  build_sdist:
+    name: Build sdist
+    runs-on: ubuntu-latest
+    needs: macos_wheels
+    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 twine
+          python -m build --sdist
+          python -m twine check dist/*
+      - uses: actions/upload-artifact@v4
+        with:
+          name: sdist
+          path: dist/*.tar.gz

{tensorstudio-0.1.1 → tensorstudio-1.0.0}/.gitignore

@@ -1,5 +1,8 @@
 build/
+build-local-check-dist/
 dist/
+wheel_artifacts/
+downloaded-artifacts/
 *.egg-info/
 __pycache__/
 *.py[cod]
@@ -13,6 +16,8 @@ __pycache__/
 htmlcov/
 site/
 .venv/
+.venv*/
+.release-check/
 venv/
 .env
 .env.*

tensorstudio-1.0.0/CHANGELOG.md

@@ -0,0 +1,53 @@
+# Changelog
+
+## 1.0.0
+
+- Promoted the CPU-only stable API foundation from release candidate status.
+- Added direct C++ buffer copies for NumPy import/export paths.
+- Added faster typed tensor construction, flat-vector export, contiguous tensor
+  copy, and floating-point optimizer update paths.
+- Added a `float32` contiguous matmul fast path that avoids double accumulation
+  and per-element dtype conversion for the common training dtype.
+- Added typed contiguous reduction kernels for `sum` and `mean`.
+- Expanded benchmark reports with framework version metadata and PyTorch
+  win/loss summaries.
+- Updated documentation to describe honest `1.0.0` performance boundaries.
+
+## 1.0.0rc2
+
+- Added `*_like` tensor creation helpers for zero, one, full, empty, uniform
+  random, and normal random tensors.
+- Added `nn.Identity`, `nn.LeakyReLU`, `nn.Softplus`,
+  `nn.BCEWithLogitsLoss`, `nn.HuberLoss`, and matching functional helpers.
+- Expanded `nn.Module` with child/module introspection, trainable parameter
+  filtering, freeze/unfreeze helpers, parameter counting, recursive `apply`,
+  and richer `repr` details.
+- Added optimizer utilities for gradient clipping and lightweight
+  `StepLR`/`ExponentialLR` learning-rate schedulers.
+- Added `len(DataLoader)` batch counting with `drop_last` support.
+- Added typed C++ fast paths for contiguous elementwise/unary kernels and a
+  contiguous matmul kernel that avoids per-element Tensor method dispatch.
+- Bumped the next release candidate after public `1.0.0rc1` artifacts.
+
+## 1.0.0rc1
+
+- Release-candidate hardening toward a CPU-only v1.0.0 API foundation.
+- Added additional tensor creation helpers, math operations, comparisons,
+  no-grad mode, state APIs, data utilities, and Windows-first CI/release docs.
+- This is not final v1.0.0; remaining platform checklist items must pass before
+  a stable release.
+
+## 0.1.1
+
+- Publishable wheel release for CPython 3.10-3.13.
+- Fixed MSVC build portability by removing non-portable `ssize_t` usage.
+- Updated wheel CI to build 64-bit Windows wheels and modern Linux wheels.
+- Expanded documentation with MkDocs navigation and detailed usage guides.
+
+## 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.1 → tensorstudio-1.0.0}/CMakeLists.txt

@@ -1,7 +1,9 @@
 cmake_minimum_required(VERSION 3.18)
 
-project(tensorstudio VERSION 0.1.1 LANGUAGES CXX)
+project(tensorstudio VERSION 1.0.0 LANGUAGES CXX)
 
+find_package(Python COMPONENTS Interpreter Development.Module REQUIRED)
+set(PYBIND11_FINDPYTHON ON)
 find_package(pybind11 CONFIG REQUIRED)
 
 set(TENSORSTUDIO_CORE_SOURCES