pixpoint 8.0.0__tar.gz

pixpoint-8.0.0/LICENSE

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

pixpoint-8.0.0/PKG-INFO

@@ -0,0 +1,169 @@
+Metadata-Version: 2.4
+Name: pixpoint
+Version: 8.0.0
+Summary: Two-point, three-point, integrated three-point and one-point statistics of HEALPix maps, on CPUs and GPUs
+Author-email: David Gebauer <git@gebauer.ai>
+License: MIT
+Project-URL: Homepage, https://github.com/D-Gebauer/PixPoint
+Project-URL: Repository, https://github.com/D-Gebauer/PixPoint
+Project-URL: Documentation, https://pixpoint.readthedocs.io
+Project-URL: Bug Tracker, https://github.com/D-Gebauer/PixPoint/issues
+Keywords: astronomy,cosmology,correlation-functions,three-point-functions,weak-lensing,galaxy-clustering,galaxy-galaxy-lensing,healpix,gpu
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Astronomy
+Classifier: Topic :: Scientific/Engineering :: Physics
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: scipy>=1.9.0
+Requires-Dist: numba>=0.50.0
+Requires-Dist: healpy>=1.10.0
+Requires-Dist: h5py>=3.0.0
+Requires-Dist: tqdm
+Provides-Extra: gpu
+Requires-Dist: cupy-cuda12x>=13.0.0; extra == "gpu"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: pytest-env>=0.8.0; extra == "dev"
+Requires-Dist: black>=22.0.0; extra == "dev"
+Requires-Dist: isort>=5.0.0; extra == "dev"
+Requires-Dist: flake8>=5.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: scipy-stubs; extra == "dev"
+Requires-Dist: pre-commit>=2.0.0; extra == "dev"
+Requires-Dist: treecorr; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: sphinx>=7.0; extra == "docs"
+Requires-Dist: myst-parser>=3.0; extra == "docs"
+Requires-Dist: furo>=2024.1; extra == "docs"
+Requires-Dist: sphinx-copybutton>=0.5; extra == "docs"
+Requires-Dist: sphinx-design>=0.6; extra == "docs"
+Dynamic: license-file
+
+# PixPoint
+
+[![CI](https://github.com/D-Gebauer/PixPoint/actions/workflows/ci.yml/badge.svg)](https://github.com/D-Gebauer/PixPoint/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/github/D-Gebauer/PixPoint/graph/badge.svg?token=F4JC08UEJP)](https://codecov.io/github/D-Gebauer/PixPoint)
+[![Documentation](https://img.shields.io/readthedocs/pixpoint/latest?logo=readthedocs&logoColor=white)](https://pixpoint.readthedocs.io)
+[![PyPI](https://img.shields.io/pypi/v/pixpoint?logo=pypi&logoColor=white)](https://pypi.org/project/pixpoint/)
+[![DOI](https://zenodo.org/badge/1161690205.svg)](https://zenodo.org/badge/latestdoi/1161690205)
+[![License: MIT](https://img.shields.io/badge/license-MIT-blue.svg)](https://github.com/D-Gebauer/PixPoint/blob/main/LICENSE)
+
+**Correlation functions of HEALPix maps, fast.**
+
+<p align="center">
+  <a href="https://pixpoint.readthedocs.io"><b>Documentation</b></a> ·
+  <a href="https://pixpoint.readthedocs.io/en/latest/quickstart.html">Getting started</a> ·
+  <a href="https://pixpoint.readthedocs.io/en/latest/install.html">Installation</a> ·
+  <a href="https://pixpoint.readthedocs.io/en/latest/citing.html">Citing</a>
+</p>
+
+PixPoint measures two-point, three-point, integrated three-point and one-point
+statistics of cosmological fields on HEALPix maps: cosmic shear, galaxy
+clustering, galaxy-galaxy lensing, convergence. You give it maps and weights,
+it gives you labelled results. The pixel pairs of a footprint are found once
+and reused for every map, so a simulation suite costs one pair search and one
+fast pass per map, on a CPU or, with the same code, on NVIDIA GPUs.
+
+```python
+from pixpoint import Global2PCF
+
+g2 = Global2PCF(nside=512, mask=mask, nbins=20, theta_min=2.5, theta_max=250)
+g2.preprocess()                          # find the pixel pairs once (save_pairs / load_pairs to reuse)
+
+res = g2.gg(shear_maps, weights)         # shear_maps (nz, 2, npix), weights (nz, npix)
+res.theta                                # bin centres, arcmin
+res.xi_p, res.xi_m                       # (n_combinations, nbins)
+res.combinations                         # which tomographic bins each row is
+
+g2.kk(delta_maps, w_cov).xi              # clustering w(theta)
+g2.kg(delta_maps, shear_maps, w_lens, w_source).xi   # galaxy-galaxy lensing gamma_t
+```
+
+## What it measures
+
+| estimator | measures | methods | result |
+|---|---|---|---|
+| `Global2PCF` | two-point functions of the whole footprint: ξ±, w(θ), γ_t | `gg`, `kk`, `kg`, `three_by_two_pt` | `TwoPointResult` |
+| `Global3PCF` | three-point functions of the whole footprint, by multipoles | `ggg`, `kkk`, `kgg`, `kkg`, `measure` | `ThreePointResult` |
+| `Integrated3PCF` | aperture masses and 2PCFs in patches: the integrated 3PCF | `gg`, `kk`, `kg`, `three_by_two_pt`, `m_a`, `m_g` | `IntegratedResult` |
+| `OnePoint` | PDFs and moments of aperture mass and smoothed scalar fields | `g`, `k` | `OnePointResult` |
+
+Method names follow TreeCorr's field codes: `g` is a shear field, `k` a scalar
+field such as a density contrast or convergence map. Every estimator is
+tomographic, and every result carries the labels of its rows.
+
+## Why PixPoint
+
+- **Exact pixel-pair estimators.** The two-point functions are ratios of
+  weighted sums over every pair of unmasked pixels, and agree with TreeCorr on
+  the pixel centres to rounding.
+- **Find pairs once, measure many maps.** A full-footprint tomographic shear
+  2PCF of DES Y3 at nside 512 (6.7 × 10⁸ pairs) takes 33 ms per map on an A100
+  after a one-off pair search; the same code runs on a laptop CPU.
+- **Everything in one vocabulary.** Shear, clustering, galaxy-galaxy lensing
+  and 3×2pt from one object; 3PCF multipoles for any mix of spin-0 and spin-2
+  fields; the integrated 3PCF with its aperture filters; aperture-mass PDFs
+  and moments.
+- **Built for suites.** Streaming HDF5 writers with provenance and resume,
+  loaders that hide the upload behind the measurement, multi-GPU splitting,
+  and an optional treecode that cuts the pair count at large separations.
+- **Checked.** Every GPU kernel has a NumPy twin and a brute-force reference
+  test at machine precision; `mypy --strict` clean; ships `py.typed`.
+
+## Installation
+
+Python 3.12 or newer:
+
+```bash
+pip install pixpoint             # CPU
+pip install "pixpoint[gpu]"      # + CuPy for NVIDIA GPUs (CUDA 12 build)
+```
+
+The `gpu` extra installs `cupy-cuda12x`, which runs on any driver that
+supports CUDA 12 or newer; for a CUDA 13 build install `pixpoint` and
+`cupy-cuda13x` instead. PixPoint picks the GPU automatically. Details, including the
+development install, in the
+[installation guide](https://pixpoint.readthedocs.io/en/latest/install.html).
+
+## Learn more
+
+- [Concepts](https://pixpoint.readthedocs.io/en/latest/concepts.html): maps
+  versus catalogues, weights and masks, what the estimators compute.
+- [Two-point functions](https://pixpoint.readthedocs.io/en/latest/global2pcf.html),
+  [three-point functions](https://pixpoint.readthedocs.io/en/latest/global3pcf.html),
+  [integrated 3PCF](https://pixpoint.readthedocs.io/en/latest/integrated3pcf.html),
+  [one-point statistics](https://pixpoint.readthedocs.io/en/latest/onepoint.html).
+- [Running on GPUs](https://pixpoint.readthedocs.io/en/latest/performance.html)
+  and [files and streaming](https://pixpoint.readthedocs.io/en/latest/io.html).
+- [Conventions](https://pixpoint.readthedocs.io/en/latest/conventions.html)
+  (shear signs, bin labels, the 3PCF projection) and the formal
+  [estimator definitions](https://pixpoint.readthedocs.io/en/latest/estimator_note.html).
+
+PixPoint was called CosmoFuse before version 8.0.0; `import CosmoFuse` is now
+`import pixpoint`, nothing else in the API changed, and old pair files are
+still read.
+
+## Citing
+
+Until PixPoint has a software paper of its own, please cite the paper it was
+developed for, Gebauer et al. 2026, *SBi3PCF: simulation-based inference with
+the integrated 3PCF*, JCAP 06 (2026) 036
+([arXiv:2510.13805](https://arxiv.org/abs/2510.13805)), and name the PixPoint
+version you used. Please also cite the papers behind the estimators you use.
+The BibTeX entry and those papers are on the
+[Citing](https://pixpoint.readthedocs.io/en/latest/citing.html) page.
+
+## Contributing
+
+Issues and pull requests are welcome; see
+[CONTRIBUTING.md](https://github.com/D-Gebauer/PixPoint/blob/main/CONTRIBUTING.md). PixPoint is released under the
+[MIT licence](https://github.com/D-Gebauer/PixPoint/blob/main/LICENSE).

pixpoint-8.0.0/README.md

@@ -0,0 +1,119 @@
+# PixPoint
+
+[![CI](https://github.com/D-Gebauer/PixPoint/actions/workflows/ci.yml/badge.svg)](https://github.com/D-Gebauer/PixPoint/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/github/D-Gebauer/PixPoint/graph/badge.svg?token=F4JC08UEJP)](https://codecov.io/github/D-Gebauer/PixPoint)
+[![Documentation](https://img.shields.io/readthedocs/pixpoint/latest?logo=readthedocs&logoColor=white)](https://pixpoint.readthedocs.io)
+[![PyPI](https://img.shields.io/pypi/v/pixpoint?logo=pypi&logoColor=white)](https://pypi.org/project/pixpoint/)
+[![DOI](https://zenodo.org/badge/1161690205.svg)](https://zenodo.org/badge/latestdoi/1161690205)
+[![License: MIT](https://img.shields.io/badge/license-MIT-blue.svg)](https://github.com/D-Gebauer/PixPoint/blob/main/LICENSE)
+
+**Correlation functions of HEALPix maps, fast.**
+
+<p align="center">
+  <a href="https://pixpoint.readthedocs.io"><b>Documentation</b></a> ·
+  <a href="https://pixpoint.readthedocs.io/en/latest/quickstart.html">Getting started</a> ·
+  <a href="https://pixpoint.readthedocs.io/en/latest/install.html">Installation</a> ·
+  <a href="https://pixpoint.readthedocs.io/en/latest/citing.html">Citing</a>
+</p>
+
+PixPoint measures two-point, three-point, integrated three-point and one-point
+statistics of cosmological fields on HEALPix maps: cosmic shear, galaxy
+clustering, galaxy-galaxy lensing, convergence. You give it maps and weights,
+it gives you labelled results. The pixel pairs of a footprint are found once
+and reused for every map, so a simulation suite costs one pair search and one
+fast pass per map, on a CPU or, with the same code, on NVIDIA GPUs.
+
+```python
+from pixpoint import Global2PCF
+
+g2 = Global2PCF(nside=512, mask=mask, nbins=20, theta_min=2.5, theta_max=250)
+g2.preprocess()                          # find the pixel pairs once (save_pairs / load_pairs to reuse)
+
+res = g2.gg(shear_maps, weights)         # shear_maps (nz, 2, npix), weights (nz, npix)
+res.theta                                # bin centres, arcmin
+res.xi_p, res.xi_m                       # (n_combinations, nbins)
+res.combinations                         # which tomographic bins each row is
+
+g2.kk(delta_maps, w_cov).xi              # clustering w(theta)
+g2.kg(delta_maps, shear_maps, w_lens, w_source).xi   # galaxy-galaxy lensing gamma_t
+```
+
+## What it measures
+
+| estimator | measures | methods | result |
+|---|---|---|---|
+| `Global2PCF` | two-point functions of the whole footprint: ξ±, w(θ), γ_t | `gg`, `kk`, `kg`, `three_by_two_pt` | `TwoPointResult` |
+| `Global3PCF` | three-point functions of the whole footprint, by multipoles | `ggg`, `kkk`, `kgg`, `kkg`, `measure` | `ThreePointResult` |
+| `Integrated3PCF` | aperture masses and 2PCFs in patches: the integrated 3PCF | `gg`, `kk`, `kg`, `three_by_two_pt`, `m_a`, `m_g` | `IntegratedResult` |
+| `OnePoint` | PDFs and moments of aperture mass and smoothed scalar fields | `g`, `k` | `OnePointResult` |
+
+Method names follow TreeCorr's field codes: `g` is a shear field, `k` a scalar
+field such as a density contrast or convergence map. Every estimator is
+tomographic, and every result carries the labels of its rows.
+
+## Why PixPoint
+
+- **Exact pixel-pair estimators.** The two-point functions are ratios of
+  weighted sums over every pair of unmasked pixels, and agree with TreeCorr on
+  the pixel centres to rounding.
+- **Find pairs once, measure many maps.** A full-footprint tomographic shear
+  2PCF of DES Y3 at nside 512 (6.7 × 10⁸ pairs) takes 33 ms per map on an A100
+  after a one-off pair search; the same code runs on a laptop CPU.
+- **Everything in one vocabulary.** Shear, clustering, galaxy-galaxy lensing
+  and 3×2pt from one object; 3PCF multipoles for any mix of spin-0 and spin-2
+  fields; the integrated 3PCF with its aperture filters; aperture-mass PDFs
+  and moments.
+- **Built for suites.** Streaming HDF5 writers with provenance and resume,
+  loaders that hide the upload behind the measurement, multi-GPU splitting,
+  and an optional treecode that cuts the pair count at large separations.
+- **Checked.** Every GPU kernel has a NumPy twin and a brute-force reference
+  test at machine precision; `mypy --strict` clean; ships `py.typed`.
+
+## Installation
+
+Python 3.12 or newer:
+
+```bash
+pip install pixpoint             # CPU
+pip install "pixpoint[gpu]"      # + CuPy for NVIDIA GPUs (CUDA 12 build)
+```
+
+The `gpu` extra installs `cupy-cuda12x`, which runs on any driver that
+supports CUDA 12 or newer; for a CUDA 13 build install `pixpoint` and
+`cupy-cuda13x` instead. PixPoint picks the GPU automatically. Details, including the
+development install, in the
+[installation guide](https://pixpoint.readthedocs.io/en/latest/install.html).
+
+## Learn more
+
+- [Concepts](https://pixpoint.readthedocs.io/en/latest/concepts.html): maps
+  versus catalogues, weights and masks, what the estimators compute.
+- [Two-point functions](https://pixpoint.readthedocs.io/en/latest/global2pcf.html),
+  [three-point functions](https://pixpoint.readthedocs.io/en/latest/global3pcf.html),
+  [integrated 3PCF](https://pixpoint.readthedocs.io/en/latest/integrated3pcf.html),
+  [one-point statistics](https://pixpoint.readthedocs.io/en/latest/onepoint.html).
+- [Running on GPUs](https://pixpoint.readthedocs.io/en/latest/performance.html)
+  and [files and streaming](https://pixpoint.readthedocs.io/en/latest/io.html).
+- [Conventions](https://pixpoint.readthedocs.io/en/latest/conventions.html)
+  (shear signs, bin labels, the 3PCF projection) and the formal
+  [estimator definitions](https://pixpoint.readthedocs.io/en/latest/estimator_note.html).
+
+PixPoint was called CosmoFuse before version 8.0.0; `import CosmoFuse` is now
+`import pixpoint`, nothing else in the API changed, and old pair files are
+still read.
+
+## Citing
+
+Until PixPoint has a software paper of its own, please cite the paper it was
+developed for, Gebauer et al. 2026, *SBi3PCF: simulation-based inference with
+the integrated 3PCF*, JCAP 06 (2026) 036
+([arXiv:2510.13805](https://arxiv.org/abs/2510.13805)), and name the PixPoint
+version you used. Please also cite the papers behind the estimators you use.
+The BibTeX entry and those papers are on the
+[Citing](https://pixpoint.readthedocs.io/en/latest/citing.html) page.
+
+## Contributing
+
+Issues and pull requests are welcome; see
+[CONTRIBUTING.md](https://github.com/D-Gebauer/PixPoint/blob/main/CONTRIBUTING.md). PixPoint is released under the
+[MIT licence](https://github.com/D-Gebauer/PixPoint/blob/main/LICENSE).

pixpoint-8.0.0/pyproject.toml

@@ -0,0 +1,132 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "pixpoint"
+version = "8.0.0"
+authors = [
+    { name="David Gebauer", email="git@gebauer.ai" },
+]
+description = "Two-point, three-point, integrated three-point and one-point statistics of HEALPix maps, on CPUs and GPUs"
+readme = "README.md"
+license = {text = "MIT"}
+requires-python = ">=3.12"
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Science/Research",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Scientific/Engineering :: Astronomy",
+    "Topic :: Scientific/Engineering :: Physics",
+]
+keywords = [
+    "astronomy", "cosmology", "correlation-functions", "three-point-functions",
+    "weak-lensing", "galaxy-clustering", "galaxy-galaxy-lensing", "healpix", "gpu",
+]
+dependencies = [
+    "numpy>=1.20.0",
+    "scipy>=1.9.0",
+    "numba>=0.50.0",
+    "healpy>=1.10.0",
+    "h5py>=3.0.0",
+    "tqdm",
+]
+
+[project.optional-dependencies]
+gpu = [
+    # The CUDA 12 wheel: plain "cupy" is the source distribution and compiles
+    # CuPy on install. CUDA 13 users install cupy-cuda13x themselves.
+    "cupy-cuda12x>=13.0.0",
+]
+dev = [
+    "pytest>=7.0.0",
+    "pytest-cov>=4.0.0",
+    "pytest-env>=0.8.0",
+    "black>=22.0.0",
+    "isort>=5.0.0",
+    "flake8>=5.0.0",
+    "mypy>=1.0.0",
+    "scipy-stubs",
+    "pre-commit>=2.0.0",
+    "treecorr",
+]
+docs = [
+    "sphinx>=7.0",
+    "myst-parser>=3.0",
+    "furo>=2024.1",
+    "sphinx-copybutton>=0.5",
+    "sphinx-design>=0.6",
+]
+
+[project.urls]
+Homepage = "https://github.com/D-Gebauer/PixPoint"
+Repository = "https://github.com/D-Gebauer/PixPoint"
+Documentation = "https://pixpoint.readthedocs.io"
+"Bug Tracker" = "https://github.com/D-Gebauer/PixPoint/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-dir]
+"" = "src"
+
+[tool.setuptools.package-data]
+"pixpoint" = ["cuda/*.cu", "cuda/*.cuh", "py.typed"]
+
+[tool.black]
+line-length = 88
+target-version = ['py312']
+include = '\.pyi?$'
+extend-exclude = '''
+/(
+  # directories
+  \.eggs
+  | \.git
+  | \.hg
+  | \.mypy_cache
+  | \.tox
+  | \.venv
+  | build
+  | dist
+)/
+'''
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3
+line_length = 88
+known_first_party = ["pixpoint"]
+
+[tool.mypy]
+python_version = "3.12"
+strict = true
+warn_unreachable = true
+
+# No stubs or py.typed marker: these are typed as Any.
+[[tool.mypy.overrides]]
+module = ["cupy", "cupy.*", "cupyx", "cupyx.*", "h5py", "h5py.*", "healpy", "healpy.*", "tqdm", "tqdm.*"]
+ignore_missing_imports = true
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+env = [
+    "NUMBA_DISABLE_JIT=1",
+]
+# Coverage is a CI concern, not a default: forcing it here made every
+# invocation -- including a single-test debug run -- instrument the whole
+# package and rewrite htmlcov/.  `make coverage` and the CI job ask for it.
+addopts = [
+    "--strict-markers",
+    "--strict-config",
+]
+markers = [
+    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
+    "gpu: marks tests that require GPU",
+]

pixpoint-8.0.0/setup.cfg

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

pixpoint-8.0.0/src/pixpoint/__init__.py

@@ -0,0 +1,86 @@
+"""
+PixPoint: GPU/CPU measurement of integrated, two- and three-point correlations.
+
+Three estimators on HEALPix maps, named after the statistic they deliver:
+:class:`Integrated3PCF` (per-patch aperture masses and 2PCFs, the i3PCF
+ingredients), :class:`Global2PCF` (footprint-wide 2PCFs),
+:class:`Global3PCF` (footprint-wide 3PCF multipoles) and :class:`OnePoint`
+(PDFs and moments of aperture fields: ``g`` for the aperture mass, ``k`` for a
+filtered scalar field).  Methods are TreeCorr
+field codes (``g`` = shear, ``k`` = scalar field): ``gg``, ``kk``, ``kg``,
+``three_by_two_pt``, ``ggg``, ``kkk``, ``kgg``, ``kkg``.  Every call returns a
+labelled result (:class:`IntegratedResult`, :class:`TwoPointResult`,
+:class:`ThreePointResult`, :class:`OnePointResult`).
+"""
+
+from .correlation_helpers import (
+    Q_crittenden,
+    Q_schneider,
+    U_crittenden,
+    U_schneider,
+    U_tophat,
+    calculate_all_zetas,
+    zeta_a_g,
+    zeta_a_minus,
+    zeta_a_plus,
+    zeta_a_t,
+    zeta_g_g,
+    zeta_g_minus,
+    zeta_g_plus,
+    zeta_g_t,
+)
+from .correlations import Integrated3PCF
+from .global_2pcf import Global2PCF
+from .global_3pcf import Global3PCF
+from .onepoint import OnePoint
+from .pipeline import MapFileLoader, MapLoader
+from .result_writer import OnePointWriter, ResultWriter, result_arrays
+from .results import (
+    Component,
+    IntegratedResult,
+    OnePointResult,
+    ThreePointResult,
+    TwoPointResult,
+)
+from .treecode import DEFAULT_RESOLUTION_FACTOR
+from .utils import pixel2RaDec, select_patch_centers
+from .zeta_writer import ZetaWriter
+
+__version__ = "8.0.0"
+__author__ = "David Gebauer"
+__email__ = "git@gebauer.ai"
+
+__all__ = [
+    "Component",
+    "Global2PCF",
+    "Global3PCF",
+    "Integrated3PCF",
+    "IntegratedResult",
+    "OnePoint",
+    "OnePointResult",
+    "OnePointWriter",
+    "ResultWriter",
+    "result_arrays",
+    "ThreePointResult",
+    "TwoPointResult",
+    "DEFAULT_RESOLUTION_FACTOR",
+    "MapLoader",
+    "MapFileLoader",
+    "Q_crittenden",
+    "Q_schneider",
+    "U_crittenden",
+    "U_schneider",
+    "U_tophat",
+    "calculate_all_zetas",
+    "zeta_g_plus",
+    "zeta_g_minus",
+    "zeta_a_plus",
+    "zeta_a_minus",
+    "zeta_g_g",
+    "zeta_a_g",
+    "zeta_g_t",
+    "zeta_a_t",
+    "pixel2RaDec",
+    "select_patch_centers",
+    "ZetaWriter",
+]

pixpoint-8.0.0/src/pixpoint/_numba.py

@@ -0,0 +1,23 @@
+"""Numba entry points with types mypy can check.
+
+``numba.prange`` is typed as a class without ``__iter__``; to a type checker
+it is ``range`` (which is also what it is with ``NUMBA_DISABLE_JIT=1``).  At
+run time the jitted functions see the real ``numba.prange``.
+"""
+
+from typing import TYPE_CHECKING, Any, Callable
+
+import numba
+from numba import njit
+
+if TYPE_CHECKING:
+    prange = range
+else:
+    from numba import prange
+
+#: ``numba.get_num_threads``, typed (numba < 0.68 leaves it untyped; going
+#: through ``Any`` is correct for both).
+_numba: Any = numba
+get_num_threads: Callable[[], int] = _numba.get_num_threads
+
+__all__ = ["get_num_threads", "njit", "prange"]

pixpoint-8.0.0/src/pixpoint/backend/__init__.py

@@ -0,0 +1,22 @@
+"""
+Backend abstraction layer for CPU (NumPy/Numba) and GPU (CuPy/CUDA) execution.
+
+All correlation measurement kernels exist in two forms:
+  - **CPU**: Numba @njit functions with parallel=True for multi-core execution
+  - **GPU**: CuPy ElementwiseKernels for simple per-pair operations, and
+    compiled CUDA kernels (.cu files in cuda/) for fused/vectorised
+    tomographic operations, launched through :class:`raw.TypedKernel`
+    (every launch checked against the signature parsed from the source)
+
+One module per kernel family, CPU kernel and GPU wrapper side by side:
+:mod:`.aperture` (M_ap, M_g), :mod:`.density` (ξ_g, ξ_t), :mod:`.shear`
+(ξ±), :mod:`.three_by_two` (the fused 3x2pt path), :mod:`.degrade` (treecode
+rows), :mod:`.threepoint` (step 1 of Global3PCF); :mod:`.combinations` holds
+the tomographic layouts of the tiled kernels, :mod:`.raw` compiling and
+launching, :mod:`.core` the :class:`Backend` that holds the active kernel set
+and provides to_device()/to_numpy() for data movement.
+"""
+
+from .core import Backend, get_backend, safe_divide
+
+__all__ = ["Backend", "get_backend", "safe_divide"]