stackkernels 0.1.0__tar.gz

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

@@ -0,0 +1,28 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install build tools
+        run: pip install build twine
+
+      - name: Build package
+        run: python -m build
+
+      - name: Publish to PyPI
+        env:
+          TWINE_USERNAME: __token__
+          TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}
+        run: twine upload dist/*

stackkernels-0.1.0/.github/workflows/test.yml

@@ -0,0 +1,33 @@
+name: Test
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11"]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Check package imports
+        run: python -c "import stackkernels"
+
+      - name: Run CPU-safe tests
+        run: pytest tests/ -m "not gpu"

stackkernels-0.1.0/.gitignore

@@ -0,0 +1,11 @@
+__pycache__/
+*.pyc
+*.egg-info/
+build/
+dist/
+.pytest_cache/
+.venv/
+venv/
+*.log
+benchmarks/results/*.csv
+benchmarks/results/*.png

stackkernels-0.1.0/PKG-INFO

@@ -0,0 +1,39 @@
+Metadata-Version: 2.5
+Name: stackkernels
+Version: 0.1.0
+Summary: Fast, fused Triton kernels for transformer models — PyTorch and JAX backends.
+Project-URL: Homepage, https://github.com/stackformer-labs/stackkernels
+License: MIT
+License-File: LICENSE
+Requires-Python: >=3.9
+Requires-Dist: torch<3.0,>=2.3
+Requires-Dist: triton>=2.3
+Provides-Extra: dev
+Requires-Dist: pytest; extra == 'dev'
+Requires-Dist: pytest-benchmark; extra == 'dev'
+Provides-Extra: jax
+Requires-Dist: jax-triton; extra == 'jax'
+Requires-Dist: jax>=0.4; extra == 'jax'
+Description-Content-Type: text/markdown
+
+# stackkernels
+
+Fast, fused Triton kernels for transformer models — built as the kernel backbone for [Stackformer](#).
+
+Supports **PyTorch** and **JAX**, with a shared, framework-agnostic Triton core.
+
+## Status
+🚧 Early development. Build order: RMSNorm → LayerNorm → Flash Attention → Dropout → Activations (SiLU/GELU/SwiGLU) → Cross-Entropy → RoPE.
+
+## Install
+```bash
+pip install stackkernels
+```
+
+## Quick start
+```python
+from stackkernels.torch import RMSNorm
+```
+
+## License
+MIT

stackkernels-0.1.0/README.md

@@ -0,0 +1,21 @@
+# stackkernels
+
+Fast, fused Triton kernels for transformer models — built as the kernel backbone for [Stackformer](#).
+
+Supports **PyTorch** and **JAX**, with a shared, framework-agnostic Triton core.
+
+## Status
+🚧 Early development. Build order: RMSNorm → LayerNorm → Flash Attention → Dropout → Activations (SiLU/GELU/SwiGLU) → Cross-Entropy → RoPE.
+
+## Install
+```bash
+pip install stackkernels
+```
+
+## Quick start
+```python
+from stackkernels.torch import RMSNorm
+```
+
+## License
+MIT

stackkernels-0.1.0/pyproject.toml

@@ -0,0 +1,28 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "stackkernels"
+version = "0.1.0"
+description = "Fast, fused Triton kernels for transformer models — PyTorch and JAX backends."
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.9"
+dependencies = [
+    "torch>=2.3,<3.0",
+    "triton>=2.3",
+]
+
+[project.optional-dependencies]
+jax = [
+    "jax>=0.4",
+    "jax-triton",
+]
+dev = [
+    "pytest",
+    "pytest-benchmark",
+]
+
+[project.urls]
+Homepage = "https://github.com/stackformer-labs/stackkernels"