tensor-ee 0.1.0__tar.gz

tensor_ee-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Noel Ivan Ulloa
+
+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.

tensor_ee-0.1.0/MANIFEST.in

@@ -0,0 +1,2 @@
+include build_support.py
+prune tests

tensor_ee-0.1.0/PKG-INFO

@@ -0,0 +1,172 @@
+Metadata-Version: 2.4
+Name: tensor-ee
+Version: 0.1.0
+Summary: Experimental building blocks for running pretrained PyTorch models natively in Google Earth Engine: a restricted CNN compiler (the SegWater U-Net; standard SMP Unet and UnetPlusPlus on ResNet and VGG encoders), a converter for supported attention models (SMP SegFormer-B0; SMP UPerNet with timm's Swin-T), and a planner and runner for staged exports.
+Author: Noel Ivan Ulloa
+License-Expression: MIT
+Project-URL: Repository, https://github.com/noelEOS/tensor-ee
+Project-URL: Issues, https://github.com/noelEOS/tensor-ee/issues
+Project-URL: Documentation, https://github.com/noelEOS/tensor-ee/blob/main/docs/README.md
+Project-URL: Changelog, https://github.com/noelEOS/tensor-ee/blob/main/CHANGELOG.md
+Keywords: earth-engine,pytorch,geospatial,remote-sensing,neural-networks
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: GIS
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: earthengine-api>=1.7.0
+Requires-Dist: torch>=2.10
+Requires-Dist: numpy>=1.26
+Provides-Extra: test
+Requires-Dist: pytest>=9.0.3; extra == "test"
+Provides-Extra: resnet
+Requires-Dist: torchvision>=0.20; extra == "resnet"
+Provides-Extra: onnx
+Requires-Dist: onnx>=1.22; extra == "onnx"
+Requires-Dist: onnxruntime>=1.20; extra == "onnx"
+Provides-Extra: segwater
+Requires-Dist: segmentation-models-pytorch==0.5.0; extra == "segwater"
+Provides-Extra: swin
+Requires-Dist: segmentation-models-pytorch==0.5.0; extra == "swin"
+Requires-Dist: timm==1.0.30; extra == "swin"
+Provides-Extra: plot
+Requires-Dist: matplotlib>=3.8; extra == "plot"
+Dynamic: license-file
+
+# TensorEE
+
+[![Offline tests](https://github.com/noelEOS/tensor-ee/actions/workflows/tests.yml/badge.svg?branch=main)](https://github.com/noelEOS/tensor-ee/actions/workflows/tests.yml)
+
+**Run supported PyTorch models inside Google Earth Engine, without deploying a
+model-serving endpoint.** TensorEE expresses neural-network inference as Earth Engine
+image and array operations. Larger models can run in stages, saving intermediate
+features as Earth Engine assets.
+
+![Earth Engine imagery passes to a deployed Vertex AI endpoint, or is evaluated directly with TensorEE's Earth Engine image and array operations](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/inference-location.svg)
+
+Python builds the computation graph; Earth Engine evaluates the forward pass. Weights
+are embedded constants or reads from EE table assets. Unsupported operations are refused,
+not approximated. The [Vertex AI integration](https://developers.google.com/earth-engine/guides/ee-vertex-overview)
+shown alongside it uses a separately deployed model endpoint.
+
+## A real example
+
+![Sentinel-1 radar imagery of Sun Moon Lake, with the water probability and mask computed by a trained U-Net inside Earth Engine](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/water_chip_224.png)
+
+*A trained ResNet18 U-Net on one 224 × 224 Sentinel-1 chip: Sun Moon Lake, Taiwan,
+3 January 2024. Prediction, not ground truth. [Validation](https://github.com/noelEOS/tensor-ee/blob/main/docs/validation.md) ·
+[Walkthrough](https://github.com/noelEOS/tensor-ee/blob/main/docs/water-chip.md). Contains modified Copernicus Sentinel data (2024).*
+
+## Get started
+
+- **[Run the Code Editor demo](https://github.com/noelEOS/tensor-ee/blob/main/CODE_EDITOR.md).** No install or checkpoint. The trained
+  U-Net reads public weights; you need an Earth Engine account and a
+  [registered Cloud project](https://developers.google.com/earth-engine/guides/access)
+  where you can create an image asset. The demo exports one chip, billed to your project.
+- **[Try the offline example](https://github.com/noelEOS/tensor-ee/blob/main/README.md#install).** Build a small CNN's graph locally, with no
+  credentials or cloud calls.
+- **[Convert an attention model](https://github.com/noelEOS/tensor-ee/blob/main/docs/getting-started.md#5-convert-an-attention-model).**
+  Start with random weights, inspect the conversion report, then plan the stages.
+
+## What you can run
+
+| Model or approach | Entry point | Availability |
+|---|---|---|
+| SegWater ResNet18 U-Net, 224 × 224 | [Code Editor demo](https://github.com/noelEOS/tensor-ee/blob/main/CODE_EDITOR.md); `compile_model` for CNNs | Trained EE weights are public; the Code Editor demo needs no checkpoint |
+| Fixed Swin-T/UPerNet, 224 × 224 | [Three-stage workflow](https://github.com/noelEOS/tensor-ee/blob/main/docs/swin.md) | Trained EE weights are public; this workflow also needs a local checkpoint, not yet public |
+| SMP SegFormer-B0 or UPerNet with timm's Swin encoder | [Attention converter](https://github.com/noelEOS/tensor-ee/blob/main/docs/converter.md) | Conversion examples use random weights; the report identifies settings outside the validated records |
+
+You can also compose a model from the [block catalogue](https://github.com/noelEOS/tensor-ee/blob/main/docs/building-blocks.md).
+It has 78 blocks: 16 in `tensor_ee/blocks/raster` and 62 in `tensor_ee/blocks/window`.
+The [browser demo](https://github.com/noelEOS/tensor-ee/blob/main/DEMO.md) runs either trained model with a local checkpoint;
+its `--offline` mode previews the interface without one.
+
+## Staged inference
+
+Saving intermediate features lets a model run as several bounded EE graphs and avoids
+recomputing earlier stages. In the fixed Swin workflow below, each step waits for its
+predecessor's export to complete. The decoder reads both C4 and the earlier features.
+
+![Swin inference in three execution steps: weights are read from EE table assets, intermediate features are saved as EE image assets, and both feature paths feed the decoder. Active steps are blue; completed steps and saved assets turn green with checkmarks.](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/swin-inference.gif)
+
+*Illustrative sequence, not execution timing. [Static diagram](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/swin-staged-inference.png).*
+
+Weights live in EE table assets; saved features and predictions live in EE image assets.
+The checkpoint, manifest and task state are local files. This workflow needs no GCS bucket.
+The [Swin guide](https://github.com/noelEOS/tensor-ee/blob/main/docs/swin.md) details the three steps, including the validity summary
+saved with C1–C3. Other models can use different cuts: the [converter planner](https://github.com/noelEOS/tensor-ee/blob/main/docs/converter.md#3-plan)
+chooses stages under explicit limits.
+
+## Install
+
+Python 3.11 or newer.
+
+```sh
+python -m pip install tensor-ee
+```
+
+Build a small CNN's graph offline, with no credentials or cloud calls:
+
+```sh
+python -m tensor_ee.examples.nonwater --prepare
+```
+
+Expect `"status": "offline_graph_only"` in the report.
+[Getting started](https://github.com/noelEOS/tensor-ee/blob/main/docs/getting-started.md)
+covers optional extras, PyTorch's download size and CPU-only installation, and the
+steps for running in Earth Engine.
+
+For the tests, tools, demos and checkout examples, install from source:
+
+```sh
+git clone https://github.com/noelEOS/tensor-ee.git
+cd tensor-ee
+python -m venv .venv && source .venv/bin/activate
+python -m pip install '.[test]'
+python -m pytest -q tests/test_nonwater_example.py
+```
+
+Expect `2 passed` from the example's tests.
+
+## Evidence and limits
+
+- **Experimental, with restricted support.** Only the settings in the
+  [block catalogue](https://github.com/noelEOS/tensor-ee/blob/main/docs/building-blocks.md) have validation evidence. A conversion can
+  report settings outside those records; converting them does not validate them.
+- **Numerical agreement is not model accuracy.** The [model evidence](https://github.com/noelEOS/tensor-ee/blob/main/docs/model.md)
+  records live comparisons with PyTorch and ONNX Runtime, offline graph identity,
+  and the inputs each result covers. The absolute logit target is 1e-4; segmentation
+  accuracy is not measured.
+- **Known exception.** The U-Net's synthetic upper-left impulse chip has a largest
+  absolute logit difference of 1.6510487e-4, above the target; probabilities and the
+  water mask agree ([record](https://github.com/noelEOS/tensor-ee/blob/main/docs/history/cnn/SEGWATER_PYTORCH_RESULTS.md)).
+- **EE limits still apply.** Inference uses EE computation and storage, with their costs
+  and quotas. The figures make no speed or cost comparison with Vertex AI.
+- **Weights and training data.** Both trained models' EE weights are public under
+  [CC BY 4.0](https://github.com/noelEOS/tensor-ee/blob/main/WEIGHTS-LICENSE.md); their [manifests](https://github.com/noelEOS/tensor-ee/tree/main/data/manifests/) list the assets.
+  Checkpoint files are not yet public. Training used Sentinel-1 chips with water labels
+  derived from Sentinel-2; a fuller dataset description is not yet published
+  ([provenance](https://github.com/noelEOS/tensor-ee/blob/main/docs/model.md#training-data-and-provenance)).
+
+## Documentation
+
+[Getting started](https://github.com/noelEOS/tensor-ee/blob/main/docs/getting-started.md) · [API guide](https://github.com/noelEOS/tensor-ee/blob/main/docs/api.md) ·
+[Converter guide](https://github.com/noelEOS/tensor-ee/blob/main/docs/converter.md) · [Block catalogue](https://github.com/noelEOS/tensor-ee/blob/main/docs/building-blocks.md) ·
+[All guides and evidence](https://github.com/noelEOS/tensor-ee/blob/main/docs/README.md) ·
+[Changelog](https://github.com/noelEOS/tensor-ee/blob/main/CHANGELOG.md)
+
+## Licence, citation and attribution
+
+Code and documentation: [MIT](https://github.com/noelEOS/tensor-ee/blob/main/LICENSE). Trained weights: [CC BY 4.0](https://github.com/noelEOS/tensor-ee/blob/main/WEIGHTS-LICENSE.md),
+including the checkpoints when published. Satellite data and third-party libraries keep
+their own terms. The Sentinel-1 examples contain modified Copernicus Sentinel data (2024).
+To cite TensorEE, use [CITATION.cff](https://github.com/noelEOS/tensor-ee/blob/main/CITATION.cff) and record the commit used.
+
+See [CONTRIBUTING](https://github.com/noelEOS/tensor-ee/blob/main/CONTRIBUTING.md) for contributions and [SECURITY.md](https://github.com/noelEOS/tensor-ee/blob/main/SECURITY.md) for
+private vulnerability reports.

tensor_ee-0.1.0/README.md

@@ -0,0 +1,131 @@
+# TensorEE
+
+[![Offline tests](https://github.com/noelEOS/tensor-ee/actions/workflows/tests.yml/badge.svg?branch=main)](https://github.com/noelEOS/tensor-ee/actions/workflows/tests.yml)
+
+**Run supported PyTorch models inside Google Earth Engine, without deploying a
+model-serving endpoint.** TensorEE expresses neural-network inference as Earth Engine
+image and array operations. Larger models can run in stages, saving intermediate
+features as Earth Engine assets.
+
+![Earth Engine imagery passes to a deployed Vertex AI endpoint, or is evaluated directly with TensorEE's Earth Engine image and array operations](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/inference-location.svg)
+
+Python builds the computation graph; Earth Engine evaluates the forward pass. Weights
+are embedded constants or reads from EE table assets. Unsupported operations are refused,
+not approximated. The [Vertex AI integration](https://developers.google.com/earth-engine/guides/ee-vertex-overview)
+shown alongside it uses a separately deployed model endpoint.
+
+## A real example
+
+![Sentinel-1 radar imagery of Sun Moon Lake, with the water probability and mask computed by a trained U-Net inside Earth Engine](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/water_chip_224.png)
+
+*A trained ResNet18 U-Net on one 224 × 224 Sentinel-1 chip: Sun Moon Lake, Taiwan,
+3 January 2024. Prediction, not ground truth. [Validation](https://github.com/noelEOS/tensor-ee/blob/main/docs/validation.md) ·
+[Walkthrough](https://github.com/noelEOS/tensor-ee/blob/main/docs/water-chip.md). Contains modified Copernicus Sentinel data (2024).*
+
+## Get started
+
+- **[Run the Code Editor demo](https://github.com/noelEOS/tensor-ee/blob/main/CODE_EDITOR.md).** No install or checkpoint. The trained
+  U-Net reads public weights; you need an Earth Engine account and a
+  [registered Cloud project](https://developers.google.com/earth-engine/guides/access)
+  where you can create an image asset. The demo exports one chip, billed to your project.
+- **[Try the offline example](https://github.com/noelEOS/tensor-ee/blob/main/README.md#install).** Build a small CNN's graph locally, with no
+  credentials or cloud calls.
+- **[Convert an attention model](https://github.com/noelEOS/tensor-ee/blob/main/docs/getting-started.md#5-convert-an-attention-model).**
+  Start with random weights, inspect the conversion report, then plan the stages.
+
+## What you can run
+
+| Model or approach | Entry point | Availability |
+|---|---|---|
+| SegWater ResNet18 U-Net, 224 × 224 | [Code Editor demo](https://github.com/noelEOS/tensor-ee/blob/main/CODE_EDITOR.md); `compile_model` for CNNs | Trained EE weights are public; the Code Editor demo needs no checkpoint |
+| Fixed Swin-T/UPerNet, 224 × 224 | [Three-stage workflow](https://github.com/noelEOS/tensor-ee/blob/main/docs/swin.md) | Trained EE weights are public; this workflow also needs a local checkpoint, not yet public |
+| SMP SegFormer-B0 or UPerNet with timm's Swin encoder | [Attention converter](https://github.com/noelEOS/tensor-ee/blob/main/docs/converter.md) | Conversion examples use random weights; the report identifies settings outside the validated records |
+
+You can also compose a model from the [block catalogue](https://github.com/noelEOS/tensor-ee/blob/main/docs/building-blocks.md).
+It has 78 blocks: 16 in `tensor_ee/blocks/raster` and 62 in `tensor_ee/blocks/window`.
+The [browser demo](https://github.com/noelEOS/tensor-ee/blob/main/DEMO.md) runs either trained model with a local checkpoint;
+its `--offline` mode previews the interface without one.
+
+## Staged inference
+
+Saving intermediate features lets a model run as several bounded EE graphs and avoids
+recomputing earlier stages. In the fixed Swin workflow below, each step waits for its
+predecessor's export to complete. The decoder reads both C4 and the earlier features.
+
+![Swin inference in three execution steps: weights are read from EE table assets, intermediate features are saved as EE image assets, and both feature paths feed the decoder. Active steps are blue; completed steps and saved assets turn green with checkmarks.](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/swin-inference.gif)
+
+*Illustrative sequence, not execution timing. [Static diagram](https://raw.githubusercontent.com/noelEOS/tensor-ee/main/docs/images/swin-staged-inference.png).*
+
+Weights live in EE table assets; saved features and predictions live in EE image assets.
+The checkpoint, manifest and task state are local files. This workflow needs no GCS bucket.
+The [Swin guide](https://github.com/noelEOS/tensor-ee/blob/main/docs/swin.md) details the three steps, including the validity summary
+saved with C1–C3. Other models can use different cuts: the [converter planner](https://github.com/noelEOS/tensor-ee/blob/main/docs/converter.md#3-plan)
+chooses stages under explicit limits.
+
+## Install
+
+Python 3.11 or newer.
+
+```sh
+python -m pip install tensor-ee
+```
+
+Build a small CNN's graph offline, with no credentials or cloud calls:
+
+```sh
+python -m tensor_ee.examples.nonwater --prepare
+```
+
+Expect `"status": "offline_graph_only"` in the report.
+[Getting started](https://github.com/noelEOS/tensor-ee/blob/main/docs/getting-started.md)
+covers optional extras, PyTorch's download size and CPU-only installation, and the
+steps for running in Earth Engine.
+
+For the tests, tools, demos and checkout examples, install from source:
+
+```sh
+git clone https://github.com/noelEOS/tensor-ee.git
+cd tensor-ee
+python -m venv .venv && source .venv/bin/activate
+python -m pip install '.[test]'
+python -m pytest -q tests/test_nonwater_example.py
+```
+
+Expect `2 passed` from the example's tests.
+
+## Evidence and limits
+
+- **Experimental, with restricted support.** Only the settings in the
+  [block catalogue](https://github.com/noelEOS/tensor-ee/blob/main/docs/building-blocks.md) have validation evidence. A conversion can
+  report settings outside those records; converting them does not validate them.
+- **Numerical agreement is not model accuracy.** The [model evidence](https://github.com/noelEOS/tensor-ee/blob/main/docs/model.md)
+  records live comparisons with PyTorch and ONNX Runtime, offline graph identity,
+  and the inputs each result covers. The absolute logit target is 1e-4; segmentation
+  accuracy is not measured.
+- **Known exception.** The U-Net's synthetic upper-left impulse chip has a largest
+  absolute logit difference of 1.6510487e-4, above the target; probabilities and the
+  water mask agree ([record](https://github.com/noelEOS/tensor-ee/blob/main/docs/history/cnn/SEGWATER_PYTORCH_RESULTS.md)).
+- **EE limits still apply.** Inference uses EE computation and storage, with their costs
+  and quotas. The figures make no speed or cost comparison with Vertex AI.
+- **Weights and training data.** Both trained models' EE weights are public under
+  [CC BY 4.0](https://github.com/noelEOS/tensor-ee/blob/main/WEIGHTS-LICENSE.md); their [manifests](https://github.com/noelEOS/tensor-ee/tree/main/data/manifests/) list the assets.
+  Checkpoint files are not yet public. Training used Sentinel-1 chips with water labels
+  derived from Sentinel-2; a fuller dataset description is not yet published
+  ([provenance](https://github.com/noelEOS/tensor-ee/blob/main/docs/model.md#training-data-and-provenance)).
+
+## Documentation
+
+[Getting started](https://github.com/noelEOS/tensor-ee/blob/main/docs/getting-started.md) · [API guide](https://github.com/noelEOS/tensor-ee/blob/main/docs/api.md) ·
+[Converter guide](https://github.com/noelEOS/tensor-ee/blob/main/docs/converter.md) · [Block catalogue](https://github.com/noelEOS/tensor-ee/blob/main/docs/building-blocks.md) ·
+[All guides and evidence](https://github.com/noelEOS/tensor-ee/blob/main/docs/README.md) ·
+[Changelog](https://github.com/noelEOS/tensor-ee/blob/main/CHANGELOG.md)
+
+## Licence, citation and attribution
+
+Code and documentation: [MIT](https://github.com/noelEOS/tensor-ee/blob/main/LICENSE). Trained weights: [CC BY 4.0](https://github.com/noelEOS/tensor-ee/blob/main/WEIGHTS-LICENSE.md),
+including the checkpoints when published. Satellite data and third-party libraries keep
+their own terms. The Sentinel-1 examples contain modified Copernicus Sentinel data (2024).
+To cite TensorEE, use [CITATION.cff](https://github.com/noelEOS/tensor-ee/blob/main/CITATION.cff) and record the commit used.
+
+See [CONTRIBUTING](https://github.com/noelEOS/tensor-ee/blob/main/CONTRIBUTING.md) for contributions and [SECURITY.md](https://github.com/noelEOS/tensor-ee/blob/main/SECURITY.md) for
+private vulnerability reports.

tensor_ee-0.1.0/build_support.py

@@ -0,0 +1,21 @@
+"""Setuptools build step: the wheel ships the tensor_ee package only.
+
+T01-T06 shipped a filtered research closure beside tensor_ee (``SWIN_MODULES``,
+pinned to docs/evidence/t01-runtime-imports.json). Since L06 the runtime closure
+of the library, the converter, the two reference models and the SegWater recipe
+is tensor_ee plus third-party dependencies (docs/evidence/l06-runtime-imports.json),
+and pyproject.toml finds ``tensor_ee*`` only. So nothing is filtered any more:
+this step builds tensor_ee unchanged and refuses any other package, so that a
+research folder cannot ship again by a change to the package list. This file is
+build tooling, not runtime.
+"""
+from setuptools.command.build_py import build_py
+
+PACKAGE = "tensor_ee"
+
+
+class RuntimeOnlyBuild(build_py):
+    def find_package_modules(self, package, package_dir):
+        if package != PACKAGE and not package.startswith(PACKAGE + "."):
+            raise ValueError(f"only the {PACKAGE} package ships in the wheel; refusing {package!r}")
+        return super().find_package_modules(package, package_dir)

tensor_ee-0.1.0/pyproject.toml

@@ -0,0 +1,52 @@
+[build-system]
+requires = ["setuptools>=83"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "tensor-ee"
+version = "0.1.0"
+description = "Experimental building blocks for running pretrained PyTorch models natively in Google Earth Engine: a restricted CNN compiler (the SegWater U-Net; standard SMP Unet and UnetPlusPlus on ResNet and VGG encoders), a converter for supported attention models (SMP SegFormer-B0; SMP UPerNet with timm's Swin-T), and a planner and runner for staged exports."
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [{name = "Noel Ivan Ulloa"}]
+requires-python = ">=3.11"
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Science/Research",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Scientific/Engineering :: GIS",
+]
+keywords = ["earth-engine", "pytorch", "geospatial", "remote-sensing", "neural-networks"]
+dependencies = ["earthengine-api>=1.7.0", "torch>=2.10", "numpy>=1.26"]
+
+[project.urls]
+Repository = "https://github.com/noelEOS/tensor-ee"
+Issues = "https://github.com/noelEOS/tensor-ee/issues"
+Documentation = "https://github.com/noelEOS/tensor-ee/blob/main/docs/README.md"
+Changelog = "https://github.com/noelEOS/tensor-ee/blob/main/CHANGELOG.md"
+
+[project.optional-dependencies]
+test = ["pytest>=9.0.3"]
+resnet = ["torchvision>=0.20"]
+onnx = ["onnx>=1.22", "onnxruntime>=1.20"]
+segwater = ["segmentation-models-pytorch==0.5.0"]
+swin = ["segmentation-models-pytorch==0.5.0", "timm==1.0.30"]
+plot = ["matplotlib>=3.8"]
+
+[tool.setuptools]
+include-package-data = false
+
+[tool.setuptools.cmdclass]
+build_py = "build_support.RuntimeOnlyBuild"
+
+[tool.setuptools.packages.find]
+include = ["tensor_ee*"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+pythonpath = ["."]

tensor_ee-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

tensor_ee-0.1.0/tensor_ee/__init__.py

@@ -0,0 +1,28 @@
+"""Experimental PyTorch-compatible building blocks for native Earth Engine."""
+
+from importlib.metadata import PackageNotFoundError as _PackageNotFoundError, version as _distribution_version
+
+try:
+    __version__ = _distribution_version("tensor-ee")
+except _PackageNotFoundError:
+    __version__ = "0.1.0"  # Source checkouts need no installed distribution metadata.
+
+from .comparison import PixelTensor, compare_segmentation
+from .compatibility import CompatibilityReport, check_compatibility
+from .compiler import CompiledModel, Grid, Raster, UnsupportedOperation, WeightProvider, compile_model
+from .primitives import (concatenate, conv2d, eval_dropout, from_image, gelu, leaky_relu, nearest_upsample,
+                         pool2d, relu, residual_add, sigmoid, silu, softmax, tanh)
+from .outputs import SegmentationOutputs, binary_threshold, segmentation_outputs
+from .weight_assets import FeatureCollectionWeightProvider, effective_convolutions, export_weights
+from .weight_mapping import MatchedWeightProvider
+
+__all__ = [
+    "PixelTensor", "compare_segmentation",
+    "CompatibilityReport", "check_compatibility",
+    "CompiledModel", "Grid", "Raster", "UnsupportedOperation", "WeightProvider",
+    "compile_model", "from_image", "conv2d", "pool2d", "nearest_upsample",
+    "concatenate", "residual_add", "relu", "sigmoid", "leaky_relu", "silu", "gelu", "tanh", "eval_dropout", "softmax",
+    "SegmentationOutputs", "segmentation_outputs", "binary_threshold",
+    "FeatureCollectionWeightProvider", "MatchedWeightProvider",
+    "effective_convolutions", "export_weights",
+]