Simple Rust implementation

pixi.toml

@@ -4,7 +4,16 @@ channels = ["conda-forge"]
 platforms = ["linux-64", "osx-arm64", "win-64"]
 
 [tasks]
-postinstall = "pip install --no-build-isolation --no-deps --disable-pip-version-check -e ."
+# Build the Rust extension using maturin (config in pyproject.toml [tool.maturin])
+build-rust = """
+    maturin build --release -m rust_ext/Cargo.toml \
+    && pip install --force-reinstall rust_ext/target/wheels/*.whl \
+    && cp $(python -c "import tabmat_rust_ext; print(tabmat_rust_ext.__file__.replace('__init__.py', 'tabmat_rust_ext.abi3.so'))") src/tabmat/tabmat_rust_ext/
+"""
+# Build only C++ extensions (skip Rust)
+postinstall-cpp = "pip install --no-build-isolation --no-deps --disable-pip-version-check -e ."
+# Full build: Rust extension first, then C++ extensions
+postinstall = { cmd = "pip install --no-build-isolation --no-deps --disable-pip-version-check -e .", depends-on = ["build-rust"] }
 
 [tasks.install-nightlies]
 cmd = """
@@ -88,6 +97,8 @@ mako = "*"
 setuptools = ">=62.0"
 setuptools-scm = ">=8.1"
 xsimd = "<11|>12.1"
+rust = ">=1.70"
+maturin = ">=1.0"
 [target.unix.dependencies]
 jemalloc-local = "*"
 

pyproject.toml

@@ -92,3 +92,17 @@ before-all = [
   "cmake -DCMAKE_INSTALL_PREFIX=/usr/local ..",
   "make install"
 ]
+
+# =============================================================================
+# Maturin configuration for the Rust extension
+# =============================================================================
+# The Rust extension is built separately using maturin. This configuration
+# tells maturin where to find the Cargo.toml and how to build the extension.
+# Build with: maturin build --release -m rust_ext/Cargo.toml
+# =============================================================================
+
+[tool.maturin]
+# Path to the Cargo.toml for the Rust extension
+manifest-path = "rust_ext/Cargo.toml"
+# Use stable ABI for Python 3.10+
+features = ["pyo3/extension-module", "pyo3/abi3-py310"]

rust_ext/Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "tabmat_rust_ext"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+name = "tabmat_rust_ext"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.23", features = ["extension-module", "abi3-py310"] }
+numpy = "0.23"
+
+[profile.release]
+lto = true
+opt-level = 3