gmtorch 0.0.0.dev0__tar.gz

gmtorch-0.0.0.dev0/.copier-answers.yml

@@ -0,0 +1,12 @@
+_commit: v1.6.0
+_src_path: gh:superlinear-ai/substrate
+author_email: tomass.timmermans@student.kit.edu
+author_name: Tomass Marks Timmermans
+project_description: Probabilistic Generative AI Using Gaussian Mixture
+project_name: gmtorch
+project_type: package
+project_url: https://gitlab.kit.edu/ali.darijani/gmtorch
+python_version: "3.12"
+typing: strict
+with_conventional_commits: true
+with_typer_cli: false

gmtorch-0.0.0.dev0/.devcontainer/devcontainer.json

@@ -0,0 +1,80 @@
+{
+    "name": "gmtorch",
+    "dockerComposeFile": "../docker-compose.yml",
+    "service": "devcontainer",
+    "workspaceFolder": "/workspaces/${localWorkspaceFolderBasename}/",
+    "features": {
+        "ghcr.io/devcontainers-extra/features/starship:1": {}
+    },
+    "overrideCommand": true,
+    "remoteUser": "user",
+    "postStartCommand": "sudo chown -R user:user /opt/ && uv sync --python ${localEnv:PYTHON_VERSION:3.12} --resolution ${localEnv:RESOLUTION_STRATEGY:highest} --all-extras && pre-commit install --install-hooks",
+    "customizations": {
+        "jetbrains": {
+            "backend": "PyCharm",
+            "plugins": [
+                "com.github.copilot"
+            ]
+        },
+        "vscode": {
+            "extensions": [
+                "charliermarsh.ruff",
+                "GitHub.copilot",
+                "GitHub.copilot-chat",
+                "GitLab.gitlab-workflow",
+                "ms-azuretools.vscode-docker",
+                "ms-python.mypy-type-checker",
+                "ms-python.python",
+                "ms-toolsai.jupyter",
+                "ryanluker.vscode-coverage-gutters",
+                "tamasfe.even-better-toml",
+                "visualstudioexptteam.vscodeintellicode"
+            ],
+            "settings": {
+                "coverage-gutters.coverageFileNames": [
+                    "reports/coverage.xml"
+                ],
+                "editor.codeActionsOnSave": {
+                    "source.fixAll": "explicit",
+                    "source.organizeImports": "explicit"
+                },
+                "editor.formatOnSave": true,
+                "[python]": {
+                    "editor.defaultFormatter": "charliermarsh.ruff"
+                },
+                "[toml]": {
+                    "editor.formatOnSave": false
+                },
+                "editor.rulers": [
+                    100
+                ],
+                "files.autoSave": "onFocusChange",
+                "github.copilot.chat.agent.enabled": true,
+                "github.copilot.chat.codesearch.enabled": true,
+                "github.copilot.chat.edits.enabled": true,
+                "github.copilot.nextEditSuggestions.enabled": true,
+                "jupyter.kernels.excludePythonEnvironments": [
+                    "/usr/local/bin/python"
+                ],
+                "mypy-type-checker.importStrategy": "fromEnvironment",
+                "mypy-type-checker.preferDaemon": true,
+                "notebook.codeActionsOnSave": {
+                    "notebook.source.fixAll": "explicit",
+                    "notebook.source.organizeImports": "explicit"
+                },
+                "notebook.formatOnSave.enabled": true,
+                "python.defaultInterpreterPath": "/opt/venv/bin/python",
+                "python.terminal.activateEnvironment": false,
+                "python.testing.pytestEnabled": true,
+                "ruff.importStrategy": "fromEnvironment",
+                "ruff.logLevel": "warning",
+                "terminal.integrated.env.linux": {
+                    "GIT_EDITOR": "code --wait"
+                },
+                "terminal.integrated.env.mac": {
+                    "GIT_EDITOR": "code --wait"
+                }
+            }
+        }
+    }
+}

gmtorch-0.0.0.dev0/.dockerignore

@@ -0,0 +1,8 @@
+# Caches
+.*_cache/
+
+# Git
+.git/
+
+# Python
+.venv/

gmtorch-0.0.0.dev0/.gitignore

@@ -0,0 +1,72 @@
+# Coverage.py
+htmlcov/
+reports/
+
+# Copier
+*.rej
+
+# Data
+*.csv*
+*.dat*
+*.pickle*
+*.xls*
+*.zip*
+data/
+
+# direnv
+.envrc
+
+# dotenv
+.env
+
+# Hypothesis
+.hypothesis/
+
+# Jupyter
+*.ipynb
+.ipynb_checkpoints/
+notebooks/
+
+# macOS
+.DS_Store
+
+# mise
+mise.local.toml
+
+# MkDocs
+site/
+
+# mypy
+.dmypy.json
+.mypy_cache/
+
+# Node.js
+node_modules/
+
+# PyCharm
+.idea/
+
+# pyenv
+.python-version
+
+# pytest
+.pytest_cache/
+
+# Python
+__pycache__/
+*.egg-info/
+*.py[cdo]
+.venv/
+dist/
+
+# Ruff
+.ruff_cache/
+
+# Terraform
+.terraform/
+
+# uv
+uv.lock
+
+# VS Code
+.vscode/

gmtorch-0.0.0.dev0/.gitlab-ci.yml

@@ -0,0 +1,338 @@
+# You can override the included template(s) by including variable overrides
+# SAST customization: https://docs.gitlab.com/ee/user/application_security/sast/#customizing-the-sast-settings
+# Secret Detection customization: https://docs.gitlab.com/user/application_security/secret_detection/pipeline/configure
+# Dependency Scanning customization: https://docs.gitlab.com/ee/user/application_security/dependency_scanning/#customizing-the-dependency-scanning-settings
+# Container Scanning customization: https://docs.gitlab.com/ee/user/application_security/container_scanning/#customizing-the-container-scanning-settings
+# Note that environment variables can be set in several places
+# See https://docs.gitlab.com/ee/ci/variables/#cicd-variable-precedence
+stages:
+  - secret-detection
+  - build
+  - test
+  - publish
+
+include:
+  - template: Security/SAST.gitlab-ci.yml
+  - template: Security/Secret-Detection.gitlab-ci.yml
+
+variables:
+  SECRET_DETECTION_ENABLED: "true"
+  GIT_STRATEGY: clone
+  GIT_CLEAN_FLAGS: -ffdx
+
+# Set the tag to the runner you want to use (also works for individual jobs)
+# KIT runners: https://docs.gitlab.kit.edu/en/gitlab_runner/
+default:
+  tags:
+    - local
+
+# SECRET DETECTION
+secret_detection:
+  stage: secret-detection
+
+# BUILD
+build_image:
+  stage: build
+  image:
+    name: gcr.io/kaniko-project/executor:debug
+    entrypoint: [""]
+  variables:
+    DOCKER_CONFIG: "/kaniko/.docker"
+    IMAGE_TAG: "$DOCKERHUB_REPO:$CI_COMMIT_SHORT_SHA"
+    DOCKERFILE_PATH: "$CI_PROJECT_DIR/Dockerfile"
+    CONTEXT: "$CI_PROJECT_DIR"
+    CACHE_REPO: "$DOCKERHUB_REPO"
+  before_script:
+    - mkdir -p /kaniko/.docker
+    - |
+      if [ -z "${DOCKERHUB_USERNAME}" ] || [ -z "${DOCKERHUB_PASSWORD}" ]; then
+        echo "DOCKERHUB_USERNAME or DOCKERHUB_PASSWORD is not set. Please define them in CI/CD variables." >&2
+        exit 1
+      fi
+      cat > /kaniko/.docker/config.json <<EOF
+      {
+        "auths": {
+          "https://index.docker.io/v1/": {
+            "auth": "$(echo -n ${DOCKERHUB_USERNAME}:${DOCKERHUB_PASSWORD} | base64)"
+          }
+        }
+      }
+      EOF
+  script:
+    - |
+      echo "Building and pushing image: $IMAGE_TAG via Kaniko"
+      /kaniko/executor \
+        --context "$CONTEXT" \
+        --dockerfile "$DOCKERFILE_PATH" \
+        --destination "$IMAGE_TAG" \
+        --target base \
+        --cache=true \
+        --cache-repo "$CACHE_REPO" \
+        --snapshot-mode=redo \
+        --build-arg BUILDKIT_INLINE_CACHE=1 \
+        --skip-unused-stages \
+        --compressed-caching=false
+
+build_project:
+  stage: build
+  image: python:3.12-alpine
+  before_script:
+    - apk add --no-cache git curl jq
+    - pip install build
+  script:
+    - |
+      set -e # Exit on error
+
+      BASE_VERSION=$(grep '^version = ' pyproject.toml | head -1 | cut -d'"' -f2)
+      PACKAGE_NAME=$(grep '^name = ' pyproject.toml | head -1 | cut -d'"' -f2)
+      BUILD_VERSION="$BASE_VERSION"
+
+      echo "Fetching existing versions from PyPI..."
+      PYPI_RESPONSE=$(curl -s "https://pypi.org/pypi/$PACKAGE_NAME/json" 2>/dev/null || echo '{"releases":{}}')
+
+      if echo "$PYPI_RESPONSE" | jq -e '.releases' > /dev/null 2>&1; then
+        PACKAGE_EXISTS=true
+        echo "✅ Package found on PyPI"
+      else
+        PACKAGE_EXISTS=false
+        echo "ℹ️  Package not yet published on PyPI"
+      fi
+      echo ""
+
+
+      if [ "$CI_COMMIT_BRANCH" = "main" ] || [ "$CI_MERGE_REQUEST_TARGET_BRANCH_NAME" = "main" ]; then
+        echo "Building stable version: $BUILD_VERSION"
+
+        if [ "$PACKAGE_EXISTS" = "true" ] && echo "$PYPI_RESPONSE" | jq -e ".releases.\"$BUILD_VERSION\"" > /dev/null 2>&1; then
+          echo "❌ ERROR: Version $BUILD_VERSION already exists on PyPI!"
+          echo ""
+          echo "Please bump the version before merging to main:"
+          echo "--> cz bump"
+          exit 1
+        fi
+        
+      elif [ "$CI_COMMIT_BRANCH" = "develop" ]; then
+        echo "Checking existing dev versions on PyPI..."
+        
+        if [ "$PACKAGE_EXISTS" = "true" ]; then
+          EXISTING_VERSIONS=$(echo "$PYPI_RESPONSE" | jq -r '.releases | keys[]' | grep "^${BASE_VERSION}\.dev" || echo "")
+        else
+          EXISTING_VERSIONS=""
+        fi
+        
+        if [ -z "$EXISTING_VERSIONS" ]; then
+          DEV_NUMBER=0
+          echo "No existing dev versions found, starting with dev0"
+        else
+          echo "Existing dev versions:"
+          echo "$EXISTING_VERSIONS"
+          
+          HIGHEST_DEV=$(echo "$EXISTING_VERSIONS" | sed "s/${BASE_VERSION}\.dev//" | sort -n | tail -1)
+          DEV_NUMBER=$((HIGHEST_DEV + 1))
+          echo "Highest existing: dev${HIGHEST_DEV}"
+          echo "Using: dev${DEV_NUMBER}"
+        fi
+        
+        BUILD_VERSION="${BASE_VERSION}.dev${DEV_NUMBER}"
+        echo "Building development version: $BUILD_VERSION"
+
+      else
+        SHORT_SHA=$(git rev-parse --short=8 HEAD)
+        BUILD_VERSION="${BASE_VERSION}.dev0+${SHORT_SHA}"
+        echo "Building feature branch version: $BUILD_VERSION"
+        echo "⚠️  Note: This version should NOT be published"
+      fi
+
+      sed -i "s/^version = .*/version = \"$BUILD_VERSION\"/" pyproject.toml
+
+      echo "Updated pyproject.toml version:"
+      grep '^version = ' pyproject.toml
+      echo ""
+
+      echo "Building package..."
+      python -m build --sdist --wheel --outdir dist
+
+      echo "BUILD_VERSION=${BUILD_VERSION}" >> build.env
+      echo "PACKAGE_NAME=${PACKAGE_NAME}" >> build.env
+
+      echo ""
+      echo "✅ Successfully built version: $BUILD_VERSION"
+      echo ""
+
+  artifacts:
+    when: on_success
+    expire_in: 1 week
+    reports:
+      dotenv: build.env
+    paths:
+      - dist/*.whl
+      - dist/*.tar.gz
+
+# TEST
+sast:
+  stage: test
+
+# All tests but cuda tests
+unit_tests:
+  stage: test
+  image: python:3.12-slim
+  needs: ["build_project"]
+  before_script:
+    - pip install pytest pytest-xdist
+    - pip install dist/*.whl
+  script:
+    - pytest tests/ --ignore=tests/cuda/ -n auto --capture=tee-sys --full-trace -v -ra --junitxml=junit.xml
+  artifacts:
+    when: on_failure
+    expire_in: 1 week
+    reports:
+      junit: junit.xml
+    paths:
+      - junit.xml
+
+# Tests run with GPU / CUDA
+gpu_tests:
+  stage: test
+  image: $DOCKERHUB_REPO:$CI_COMMIT_SHORT_SHA
+  needs: ["build_image"]
+  # Install pytest-xdist and add -n auto for parallel execution if needed
+  before_script:
+    - uv pip install pytest
+  script:
+    - pytest tests/cuda/ -v --junitxml=junit-gpu.xml
+  artifacts:
+    when: on_failure
+    expire_in: 1 week
+    reports:
+      junit: junit-gpu.xml
+    paths:
+      - junit-gpu.xml
+
+# PUBLISH
+publish_artifacts:
+  stage: publish
+  image: python:3.12-alpine
+  needs:
+    - job: build_project
+      artifacts: true
+    - job: unit_tests
+    - job: gpu_tests
+  when: manual
+  before_script:
+    - pip install twine
+  script:
+    - |
+      echo "Publishing version $BUILD_VERSION to GitLab Package Registry"
+      echo "Package: $PACKAGE_NAME"
+      TWINE_PASSWORD=${CI_JOB_TOKEN} TWINE_USERNAME=gitlab-ci-token python -m twine upload --repository-url ${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/packages/pypi dist/*.whl dist/*.tar.gz
+  rules:
+    - if: $CI_COMMIT_BRANCH == "main" || $CI_COMMIT_BRANCH == "develop"
+
+publish_test_pypi:
+  stage: publish
+  image: python:3.12-alpine
+  needs:
+    - job: build_project
+      artifacts: true
+    - job: unit_tests
+    - job: gpu_tests
+  when: manual
+  before_script:
+    - pip install twine
+  script:
+    - |
+      echo "Publishing version $BUILD_VERSION to Test PyPI"
+      python -m twine upload --repository testpypi dist/*.whl dist/*.tar.gz
+  variables:
+    TWINE_USERNAME: __token__
+    TWINE_PASSWORD: ${TEST_PYPI_API_TOKEN}
+    TWINE_REPOSITORY_URL: https://test.pypi.org/legacy/
+  rules:
+    - if: $CI_COMMIT_BRANCH == "main" || $CI_COMMIT_BRANCH == "develop"
+
+publish_pypi:
+  stage: publish
+  image: python:3.12-alpine
+  needs:
+    - job: build_project
+      artifacts: true
+    - job: unit_tests
+    - job: gpu_tests
+  when: manual
+  before_script:
+    - pip install twine
+  script:
+    - |
+      echo "Publishing version $BUILD_VERSION to PyPI" 
+      python -m twine upload dist/*.whl dist/*.tar.gz
+  variables:
+    TWINE_USERNAME: __token__
+    TWINE_PASSWORD: ${PYPI_API_TOKEN}
+  rules:
+    - if: $CI_COMMIT_BRANCH == "main" || $CI_COMMIT_BRANCH == "develop"
+
+publish_docker:
+  stage: publish
+  image:
+    name: gcr.io/kaniko-project/executor:debug
+    entrypoint: [""]
+  needs:
+    - job: build_project
+      artifacts: true
+    - job: build_image
+    - job: unit_tests
+    - job: gpu_tests
+  when: manual
+  variables:
+    DOCKER_CONFIG: "/kaniko/.docker"
+    DOCKERFILE_PATH: "$CI_PROJECT_DIR/Dockerfile"
+    CONTEXT: "$CI_PROJECT_DIR"
+    CACHE_REPO: "$DOCKERHUB_REPO"
+  before_script:
+    - mkdir -p /kaniko/.docker
+    - |
+      if [ -z "${DOCKERHUB_USERNAME}" ] || [ -z "${DOCKERHUB_PASSWORD}" ]; then
+        echo "DOCKERHUB_USERNAME or DOCKERHUB_PASSWORD is not set. Please define them in CI/CD variables." >&2
+        exit 1
+      fi
+      cat > /kaniko/.docker/config.json <<EOF
+      {
+        "auths": {
+          "https://index.docker.io/v1/": {
+            "auth": "$(echo -n ${DOCKERHUB_USERNAME}:${DOCKERHUB_PASSWORD} | base64)"
+          }
+        }
+      }
+      EOF
+  script:
+    - |
+      # Determine tags based on branch
+      if [ "$CI_COMMIT_BRANCH" = "main" ]; then
+        # Get version from pyproject.toml (stable version)
+        VERSION=$(grep '^version = ' pyproject.toml | cut -d'"' -f2)
+        echo "Publishing stable Docker image: stable"
+        /kaniko/executor \
+          --context "$CONTEXT" \
+          --dockerfile "$DOCKERFILE_PATH" \
+          --destination "$DOCKERHUB_REPO:stable" \
+          --cache=true \
+          --cache-repo "$CACHE_REPO" \
+          --compressed-caching=false \
+          --build-arg BUILDKIT_INLINE_CACHE=1 \
+          --skip-unused-stages \
+          --single-snapshot
+      elif [ "$CI_COMMIT_BRANCH" = "develop" ]; then
+        echo "Publishing development Docker image: develop"
+        /kaniko/executor \
+          --context "$CONTEXT" \
+          --dockerfile "$DOCKERFILE_PATH" \
+          --destination "$DOCKERHUB_REPO:develop" \
+          --cache=true \
+          --cache-repo "$CACHE_REPO" \
+          --compressed-caching=false \
+          --build-arg BUILDKIT_INLINE_CACHE=1 \
+          --skip-unused-stages \
+          --single-snapshot
+      fi
+  rules:
+    - if: $CI_COMMIT_BRANCH == "main" || $CI_COMMIT_BRANCH == "develop"

gmtorch-0.0.0.dev0/.pre-commit-config.yaml

@@ -0,0 +1,73 @@
+# https://pre-commit.com
+default_install_hook_types: [commit-msg, pre-commit]
+default_stages: [pre-commit, manual]
+fail_fast: true
+repos:
+  - repo: meta
+    hooks:
+      - id: check-useless-excludes
+  - repo: https://github.com/pre-commit/pygrep-hooks
+    rev: v1.10.0
+    hooks:
+      - id: python-check-mock-methods
+      - id: python-use-type-annotations
+      - id: rst-backticks
+      - id: rst-directive-colons
+      - id: rst-inline-touching-normal
+      - id: text-unicode-replacement-char
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v5.0.0
+    hooks:
+      - id: check-added-large-files
+      - id: check-ast
+      - id: check-builtin-literals
+      - id: check-case-conflict
+      - id: check-docstring-first
+      - id: check-illegal-windows-names
+      - id: check-json
+      - id: check-merge-conflict
+      - id: check-shebang-scripts-are-executable
+      - id: check-symlinks
+      - id: check-toml
+      - id: check-vcs-permalinks
+      - id: check-xml
+      - id: check-yaml
+      - id: debug-statements
+      - id: destroyed-symlinks
+      - id: detect-private-key
+      - id: end-of-file-fixer
+        types: [python]
+      - id: fix-byte-order-marker
+      - id: mixed-line-ending
+      - id: name-tests-test
+        args: [--pytest-test-first]
+      - id: trailing-whitespace
+        types: [python]
+  - repo: local
+    hooks:
+      - id: commitizen
+        name: commitizen
+        entry: cz check
+        args: [--commit-msg-file]
+        require_serial: true
+        language: system
+        stages: [commit-msg]
+      - id: ruff-check
+        name: ruff check
+        entry: ruff check
+        args: ["--force-exclude", "--extend-fixable=ERA001,F401,F841,T201,T203"]
+        require_serial: true
+        language: system
+        types_or: [python, pyi]
+      - id: ruff-format
+        name: ruff format
+        entry: ruff format
+        args: [--force-exclude]
+        require_serial: true
+        language: system
+        types_or: [python, pyi]
+      - id: mypy
+        name: mypy
+        entry: mypy
+        language: system
+        types: [python]

gmtorch-0.0.0.dev0/Dockerfile

@@ -0,0 +1,56 @@
+# syntax=docker/dockerfile:1
+
+# Build stage: Create the image with dependencies
+FROM nvidia/cuda:12.6.0-runtime-ubuntu24.04 AS base
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.12 \
+    python3.12-venv \
+    && rm -rf /var/lib/apt/lists/* \
+    && apt-get clean
+
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /usr/local/bin/uv
+
+# Set working directory
+WORKDIR /app
+
+# Setup virtual environment
+ENV VIRTUAL_ENV=/opt/venv
+ENV PATH=$VIRTUAL_ENV/bin:$PATH
+ENV UV_PROJECT_ENVIRONMENT=$VIRTUAL_ENV
+
+COPY pyproject.toml README.md ./
+COPY src ./src
+COPY tests ./tests
+
+# Create venv and install dependencies
+RUN uv venv "$VIRTUAL_ENV" && \
+    uv pip install --no-cache -e .
+
+FROM base AS dev
+
+# Install git, sudo, and create non-root user with passwordless sudo
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    sudo \
+    && rm -rf /var/lib/apt/lists/* \
+    && apt-get clean \
+    && groupadd --gid 1000 user \
+    && useradd --create-home --no-log-init --gid 1000 --uid 1000 --shell /usr/bin/bash user \
+    && chown user:user /opt/venv \
+    && echo 'user ALL=(root) NOPASSWD:ALL' > /etc/sudoers.d/user && chmod 0440 /etc/sudoers.d/user
+
+# Tell Git that the workspace is safe to avoid 'detected dubious ownership in repository' warnings.
+RUN git config --system --add safe.directory '*'
+
+USER user
+
+# Configure the non-root user's shell.
+RUN mkdir ~/.history/ && \
+    echo 'HISTFILE=~/.history/.bash_history' >> ~/.bashrc && \
+    echo 'bind "\e[A": history-search-backward' >> ~/.bashrc && \
+    echo 'bind "\e[B": history-search-forward' >> ~/.bashrc && \
+    echo 'eval "$(starship init bash)"' >> ~/.bashrc

gmtorch-0.0.0.dev0/PKG-INFO

@@ -0,0 +1,109 @@
+Metadata-Version: 2.4
+Name: gmtorch
+Version: 0.0.0.dev0
+Summary: Probabilistic Generative AI Using Gaussian Mixture
+Project-URL: homepage, https://gitlab.kit.edu/ali.darijani/gmtorch
+Project-URL: source, https://gitlab.kit.edu/ali.darijani/gmtorch
+Project-URL: changelog, https://gitlab.kit.edu/ali.darijani/gmtorch/-/blob/main/CHANGELOG.md
+Project-URL: releasenotes, https://gitlab.kit.edu/ali.darijani/gmtorch/-/releases
+Project-URL: documentation, https://gitlab.kit.edu/ali.darijani/gmtorch
+Project-URL: issues, https://gitlab.kit.edu/ali.darijani/gmtorch/-/issues
+Author-email: Ali Darijani <ali.darijani@kit.edu>, Tomass Marks Timmermans <tomass.timmermans@student.kit.edu>
+Requires-Python: <4.0,>=3.12
+Requires-Dist: torch>=2.9.0
+Requires-Dist: torchvision>=0.24.0
+Description-Content-Type: text/markdown
+
+[![Open in Dev Containers](https://img.shields.io/static/v1?label=Dev%20Containers&message=Open&color=blue&logo=data:image/svg%2bxml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHZpZXdCb3g9IjAgMCAyNCAyNCI+PHBhdGggZmlsbD0iI2ZmZiIgZD0iTTE3IDE2VjdsLTYgNU0yIDlWOGwxLTFoMWw0IDMgOC04aDFsNCAyIDEgMXYxNGwtMSAxLTQgMmgtMWwtOC04LTQgM0gzbC0xLTF2LTFsMy0zIi8+PC9zdmc+)](https://vscode.dev/redirect?url=vscode://ms-vscode-remote.remote-containers/cloneInVolume?url=https://gitlab.kit.edu/ali.darijani/gmtorch)
+
+# gmtorch
+
+Probabilistic Generative AI Using Gaussian Mixture
+
+## Installing
+
+To install this package, run:
+
+```sh
+pip install gmtorch
+```
+
+## Using
+
+Example usage:
+
+```python
+import gmtorch
+
+...
+```
+
+## Contributing
+
+<details>
+<summary>Prerequisites</summary>
+
+1. [Generate an SSH key](https://docs.gitlab.com/ee/user/ssh.html#generate-an-ssh-key-pair) and [add the SSH key to your GitLab account](https://docs.gitlab.com/ee/user/ssh.html#add-an-ssh-key-to-your-gitlab-account).
+1. Configure SSH to automatically load your SSH keys:
+
+    ```sh
+    cat << EOF >> ~/.ssh/config
+    
+    Host *
+      AddKeysToAgent yes
+      IgnoreUnknown UseKeychain
+      UseKeychain yes
+      ForwardAgent yes
+    EOF
+    ```
+
+1. [Install Docker Desktop](https://www.docker.com/get-started).
+1. [Install VS Code](https://code.visualstudio.com/) and [VS Code's Dev Containers extension](https://marketplace.visualstudio.com/items?itemName=ms-vscode-remote.remote-containers). Alternatively, install [PyCharm](https://www.jetbrains.com/pycharm/download/).
+1. _Optional:_ install a [Nerd Font](https://www.nerdfonts.com/font-downloads) such as [FiraCode Nerd Font](https://github.com/ryanoasis/nerd-fonts/tree/master/patched-fonts/FiraCode) and [configure VS Code](https://github.com/tonsky/FiraCode/wiki/VS-Code-Instructions) or [PyCharm](https://github.com/tonsky/FiraCode/wiki/Intellij-products-instructions) to use it.
+
+</details>
+
+<details open>
+<summary>Development environments</summary>
+
+The following development environments are supported:
+
+
+1. ⭐️ _VS Code Dev Container (with container volume)_: click on [Open in Dev Containers](https://vscode.dev/redirect?url=vscode://ms-vscode-remote.remote-containers/cloneInVolume?url=https://gitlab.kit.edu/ali.darijani/gmtorch) to clone this repository in a container volume and create a Dev Container with VS Code.
+1. ⭐️ _uv_: clone this repository and run the following from root of the repository:
+
+    ```sh
+    # Create and install a virtual environment
+    uv sync --python 3.12 --all-extras
+
+    # Activate the virtual environment
+    source .venv/bin/activate
+
+    # Install the pre-commit hooks
+    pre-commit install --install-hooks
+    ```
+
+1. _VS Code Dev Container_: clone this repository, open it with VS Code, and run <kbd>Ctrl/⌘</kbd> + <kbd>⇧</kbd> + <kbd>P</kbd> → _Dev Containers: Reopen in Container_.
+1. _PyCharm Dev Container_: clone this repository, open it with PyCharm, [create a Dev Container with Mount Sources](https://www.jetbrains.com/help/pycharm/start-dev-container-inside-ide.html), and [configure an existing Python interpreter](https://www.jetbrains.com/help/pycharm/configuring-python-interpreter.html#widget) at `/opt/venv/bin/python`.
+
+</details>
+
+<details open>
+<summary>Developing</summary>
+
+- This project follows the [Conventional Commits](https://www.conventionalcommits.org/) standard to automate [Semantic Versioning](https://semver.org/) and [Keep A Changelog](https://keepachangelog.com/) with [Commitizen](https://github.com/commitizen-tools/commitizen).
+- Run `poe` from within the development environment to print a list of [Poe the Poet](https://github.com/nat-n/poethepoet) tasks available to run on this project.
+- Run `uv add {package}` from within the development environment to install a run time dependency and add it to `pyproject.toml` and `uv.lock`. Add `--dev` to install a development dependency.
+- Run `uv sync --upgrade` from within the development environment to upgrade all dependencies to the latest versions allowed by `pyproject.toml`. Add `--only-dev` to upgrade the development dependencies only.
+- Run `cz bump` to bump the package's version, update the `CHANGELOG.md`, and create a git tag. Then push the changes and the git tag with `git push origin main --tags`.
+- Workflow: create a feature branch from `develop` -> create a merge request to `develop` -> on major releases, create a merge request from `develop` to `main`.
+
+</details>
+
+<details open>
+<summary>Versioning</summary>
+
+- Stable releases on `main` branch use [Semantic Versioning](https://semver.org/).
+- Development releases on `develop` branch use the current version with .devN suffix, where N is the number of published development releases so far for the current version.
+- Docker images use either `develop` or `stable` tags for publishing from `develop` or `main` branches, respectively.
+</details>

gmtorch-0.0.0.dev0/README.md

@@ -0,0 +1,93 @@
+[![Open in Dev Containers](https://img.shields.io/static/v1?label=Dev%20Containers&message=Open&color=blue&logo=data:image/svg%2bxml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHZpZXdCb3g9IjAgMCAyNCAyNCI+PHBhdGggZmlsbD0iI2ZmZiIgZD0iTTE3IDE2VjdsLTYgNU0yIDlWOGwxLTFoMWw0IDMgOC04aDFsNCAyIDEgMXYxNGwtMSAxLTQgMmgtMWwtOC04LTQgM0gzbC0xLTF2LTFsMy0zIi8+PC9zdmc+)](https://vscode.dev/redirect?url=vscode://ms-vscode-remote.remote-containers/cloneInVolume?url=https://gitlab.kit.edu/ali.darijani/gmtorch)
+
+# gmtorch
+
+Probabilistic Generative AI Using Gaussian Mixture
+
+## Installing
+
+To install this package, run:
+
+```sh
+pip install gmtorch
+```
+
+## Using
+
+Example usage:
+
+```python
+import gmtorch
+
+...
+```
+
+## Contributing
+
+<details>
+<summary>Prerequisites</summary>
+
+1. [Generate an SSH key](https://docs.gitlab.com/ee/user/ssh.html#generate-an-ssh-key-pair) and [add the SSH key to your GitLab account](https://docs.gitlab.com/ee/user/ssh.html#add-an-ssh-key-to-your-gitlab-account).
+1. Configure SSH to automatically load your SSH keys:
+
+    ```sh
+    cat << EOF >> ~/.ssh/config
+    
+    Host *
+      AddKeysToAgent yes
+      IgnoreUnknown UseKeychain
+      UseKeychain yes
+      ForwardAgent yes
+    EOF
+    ```
+
+1. [Install Docker Desktop](https://www.docker.com/get-started).
+1. [Install VS Code](https://code.visualstudio.com/) and [VS Code's Dev Containers extension](https://marketplace.visualstudio.com/items?itemName=ms-vscode-remote.remote-containers). Alternatively, install [PyCharm](https://www.jetbrains.com/pycharm/download/).
+1. _Optional:_ install a [Nerd Font](https://www.nerdfonts.com/font-downloads) such as [FiraCode Nerd Font](https://github.com/ryanoasis/nerd-fonts/tree/master/patched-fonts/FiraCode) and [configure VS Code](https://github.com/tonsky/FiraCode/wiki/VS-Code-Instructions) or [PyCharm](https://github.com/tonsky/FiraCode/wiki/Intellij-products-instructions) to use it.
+
+</details>
+
+<details open>
+<summary>Development environments</summary>
+
+The following development environments are supported:
+
+
+1. ⭐️ _VS Code Dev Container (with container volume)_: click on [Open in Dev Containers](https://vscode.dev/redirect?url=vscode://ms-vscode-remote.remote-containers/cloneInVolume?url=https://gitlab.kit.edu/ali.darijani/gmtorch) to clone this repository in a container volume and create a Dev Container with VS Code.
+1. ⭐️ _uv_: clone this repository and run the following from root of the repository:
+
+    ```sh
+    # Create and install a virtual environment
+    uv sync --python 3.12 --all-extras
+
+    # Activate the virtual environment
+    source .venv/bin/activate
+
+    # Install the pre-commit hooks
+    pre-commit install --install-hooks
+    ```
+
+1. _VS Code Dev Container_: clone this repository, open it with VS Code, and run <kbd>Ctrl/⌘</kbd> + <kbd>⇧</kbd> + <kbd>P</kbd> → _Dev Containers: Reopen in Container_.
+1. _PyCharm Dev Container_: clone this repository, open it with PyCharm, [create a Dev Container with Mount Sources](https://www.jetbrains.com/help/pycharm/start-dev-container-inside-ide.html), and [configure an existing Python interpreter](https://www.jetbrains.com/help/pycharm/configuring-python-interpreter.html#widget) at `/opt/venv/bin/python`.
+
+</details>
+
+<details open>
+<summary>Developing</summary>
+
+- This project follows the [Conventional Commits](https://www.conventionalcommits.org/) standard to automate [Semantic Versioning](https://semver.org/) and [Keep A Changelog](https://keepachangelog.com/) with [Commitizen](https://github.com/commitizen-tools/commitizen).
+- Run `poe` from within the development environment to print a list of [Poe the Poet](https://github.com/nat-n/poethepoet) tasks available to run on this project.
+- Run `uv add {package}` from within the development environment to install a run time dependency and add it to `pyproject.toml` and `uv.lock`. Add `--dev` to install a development dependency.
+- Run `uv sync --upgrade` from within the development environment to upgrade all dependencies to the latest versions allowed by `pyproject.toml`. Add `--only-dev` to upgrade the development dependencies only.
+- Run `cz bump` to bump the package's version, update the `CHANGELOG.md`, and create a git tag. Then push the changes and the git tag with `git push origin main --tags`.
+- Workflow: create a feature branch from `develop` -> create a merge request to `develop` -> on major releases, create a merge request from `develop` to `main`.
+
+</details>
+
+<details open>
+<summary>Versioning</summary>
+
+- Stable releases on `main` branch use [Semantic Versioning](https://semver.org/).
+- Development releases on `develop` branch use the current version with .devN suffix, where N is the number of published development releases so far for the current version.
+- Docker images use either `develop` or `stable` tags for publishing from `develop` or `main` branches, respectively.
+</details>

gmtorch-0.0.0.dev0/docker-compose.yml

@@ -0,0 +1,19 @@
+services:
+
+  devcontainer:
+    build:
+      target: dev
+    gpus: all
+    volumes:
+      - ..:/workspaces
+      - command-history-volume:/home/user/.history/
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [ gpu ]
+
+volumes:
+  command-history-volume:

gmtorch-0.0.0.dev0/docs/index.md

@@ -0,0 +1,3 @@
+# gmtorch
+
+Probabilistic Generative AI Using Gaussian Mixture

gmtorch-0.0.0.dev0/docs/reference.md

@@ -0,0 +1,3 @@
+# Reference
+
+::: gmtorch

gmtorch-0.0.0.dev0/gl-secret-detection-report.json

@@ -0,0 +1,28 @@
+{
+  "version": "15.2.2",
+  "vulnerabilities": [],
+  "scan": {
+    "analyzer": {
+      "id": "secrets",
+      "name": "secrets",
+      "url": "https://gitlab.com/gitlab-org/security-products/analyzers/secrets",
+      "vendor": {
+        "name": "GitLab"
+      },
+      "version": "7.18.0"
+    },
+    "scanner": {
+      "id": "gitleaks",
+      "name": "Gitleaks",
+      "url": "https://github.com/gitleaks/gitleaks",
+      "vendor": {
+        "name": "GitLab"
+      },
+      "version": "8.28.0"
+    },
+    "type": "secret_detection",
+    "start_time": "2025-10-28T15:59:43",
+    "end_time": "2025-10-28T15:59:44",
+    "status": "success"
+  }
+}

gmtorch-0.0.0.dev0/mkdocs.yml

@@ -0,0 +1,35 @@
+site_name: gmtorch
+site_description: Probabilistic Generative AI Using Gaussian Mixture
+site_url: https://gitlab.kit.edu/ali.darijani/gmtorch
+repo_url: https://gitlab.kit.edu/ali.darijani/gmtorch
+repo_name: ali.darijani/gmtorch
+
+strict: true
+
+validation:
+  omitted_files: warn
+  absolute_links: warn
+  unrecognized_links: warn
+  anchors: warn
+
+theme:
+  name: material
+
+nav:
+  - Home: index.md
+  - Reference: reference.md
+
+watch:
+  - src
+
+plugins:
+  - search
+  - mkdocstrings:
+      handlers:
+        python:
+          options:
+            docstring_style: numpy
+
+markdown_extensions:
+  - pymdownx.highlight
+  - pymdownx.superfences

gmtorch-0.0.0.dev0/pyproject.toml

@@ -0,0 +1,167 @@
+[build-system]  # https://docs.astral.sh/uv/concepts/projects/config/#build-systems
+requires = ["hatchling>=1.27.0"]
+build-backend = "hatchling.build"
+
+[project]  # https://packaging.python.org/en/latest/specifications/pyproject-toml/
+name = "gmtorch"
+version = "0.0.0.dev0"
+description = "Probabilistic Generative AI Using Gaussian Mixture"
+readme = "README.md"
+authors = [
+  { name = "Ali Darijani", email = "ali.darijani@kit.edu" },
+  { name = "Tomass Marks Timmermans", email = "tomass.timmermans@student.kit.edu" }
+
+]
+requires-python = ">=3.12,<4.0"
+dependencies = [
+    "torch>=2.9.0",
+    "torchvision>=0.24.0",
+]
+
+[tool.uv.sources]
+torchvision = [
+  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cu126"
+url = "https://download.pytorch.org/whl/cu126"
+explicit = true
+
+[project.urls]  # https://packaging.python.org/en/latest/specifications/well-known-project-urls/#well-known-labels
+homepage = "https://gitlab.kit.edu/ali.darijani/gmtorch"
+source = "https://gitlab.kit.edu/ali.darijani/gmtorch"
+changelog = "https://gitlab.kit.edu/ali.darijani/gmtorch/-/blob/main/CHANGELOG.md"
+releasenotes = "https://gitlab.kit.edu/ali.darijani/gmtorch/-/releases"
+documentation = "https://gitlab.kit.edu/ali.darijani/gmtorch"
+issues = "https://gitlab.kit.edu/ali.darijani/gmtorch/-/issues"
+
+[dependency-groups]  # https://docs.astral.sh/uv/concepts/projects/dependencies/#development-dependencies
+dev = [
+  "commitizen (>=4.3.0)",
+  "coverage[toml] (>=7.6.10)",
+  "ipykernel (>=6.29.4)",
+  "ipython (>=8.18.0)",
+  "ipywidgets (>=8.1.2)",
+  "mypy (>=1.14.1)",
+  "mkdocs-material (>=9.5.21)",
+  "mkdocstrings[python] (>=0.26.2)",
+  "poethepoet (>=0.32.1)",
+  "pre-commit (>=4.0.1)",
+  "pytest (>=8.3.4)",
+  "pytest-mock (>=3.14.0)",
+  "pytest-xdist (>=3.6.1)",
+  "ruff (>=0.9.2)",
+  "typeguard (>=4.4.1)",
+]
+
+[tool.commitizen]  # https://commitizen-tools.github.io/commitizen/config/
+bump_message = "bump: v$current_version → v$new_version"
+tag_format = "v$version"
+update_changelog_on_bump = true
+version_provider = "uv"
+
+[tool.coverage.report]  # https://coverage.readthedocs.io/en/latest/config.html#report
+fail_under = 50
+precision = 1
+show_missing = true
+skip_covered = true
+
+[tool.coverage.run]  # https://coverage.readthedocs.io/en/latest/config.html#run
+branch = true
+command_line = "--module pytest"
+data_file = "reports/.coverage"
+source = ["src"]
+
+[tool.coverage.xml]  # https://coverage.readthedocs.io/en/latest/config.html#xml
+output = "reports/coverage.xml"
+
+[tool.mypy]  # https://mypy.readthedocs.io/en/latest/config_file.html
+junit_xml = "reports/mypy.xml"
+strict = true
+disallow_subclassing_any = false
+disallow_untyped_decorators = false
+ignore_missing_imports = true
+pretty = true
+show_column_numbers = true
+show_error_codes = true
+show_error_context = true
+warn_unreachable = true
+
+[tool.pytest.ini_options]  # https://docs.pytest.org/en/latest/reference/reference.html#ini-options-ref
+addopts = "--color=yes --doctest-modules --exitfirst --failed-first --strict-config --strict-markers --verbosity=2 --junitxml=reports/pytest.xml"
+filterwarnings = ["error", "ignore::DeprecationWarning"]
+testpaths = ["src", "tests"]
+xfail_strict = true
+
+[tool.ruff]  # https://docs.astral.sh/ruff/settings/
+fix = true
+line-length = 100
+src = ["src", "tests"]
+target-version = "py312"
+
+[tool.ruff.format]
+docstring-code-format = true
+skip-magic-trailing-comma = true
+
+[tool.ruff.lint]
+select = ["ALL"]
+ignore = ["CPY", "FIX", "ARG001", "COM812", "D203", "D213", "E501", "PD008", "PD009", "RET504", "S101", "TD003"]
+unfixable = ["ERA001", "F401", "F841", "T201", "T203"]
+
+[tool.ruff.lint.flake8-annotations]
+allow-star-arg-any = true
+
+[tool.ruff.lint.flake8-tidy-imports]
+ban-relative-imports = "all"
+
+[tool.ruff.lint.isort]
+split-on-trailing-comma = false
+
+[tool.ruff.lint.pycodestyle]
+max-doc-length = 100
+
+[tool.ruff.lint.pydocstyle]
+convention = "numpy"
+
+[tool.poe.executor]  # https://github.com/nat-n/poethepoet
+type = "simple"
+
+[tool.poe.tasks]
+
+  [tool.poe.tasks.docs]
+  help = "Build or serve the documentation"
+  shell = """
+    if [ $serve ]
+    then {
+      mkdocs serve
+    } else {
+      mkdocs build
+    } fi
+    """
+
+    [[tool.poe.tasks.docs.args]]
+    help = "Serve the documentation locally with live reload"
+    type = "boolean"
+    name = "serve"
+    options = ["--serve"]
+
+  [tool.poe.tasks.lint]
+  help = "Lint this package"
+  cmd = """
+    pre-commit run
+      --all-files
+      --color always
+    """
+
+  [tool.poe.tasks.test]
+  help = "Test this package"
+
+    [[tool.poe.tasks.test.sequence]]
+    cmd = "coverage run"
+
+    [[tool.poe.tasks.test.sequence]]
+    cmd = "coverage report"
+
+    [[tool.poe.tasks.test.sequence]]
+    cmd = "coverage xml"

gmtorch-0.0.0.dev0/requirements.txt

@@ -0,0 +1,2 @@
+torch>=2.0.0
+pytest>=7.0.0

gmtorch-0.0.0.dev0/src/gmtorch/__init__.py

@@ -0,0 +1 @@
+"""gmtorch."""

gmtorch-0.0.0.dev0/tests/__init__.py

@@ -0,0 +1 @@
+"""gmtorch test suite."""

gmtorch-0.0.0.dev0/tests/cuda/__init__.py

@@ -0,0 +1 @@
+"""CUDA tests for gmtorch."""

gmtorch-0.0.0.dev0/tests/cuda/test_cuda.py

@@ -0,0 +1,23 @@
+"""Unit tests for CUDA functionality."""
+
+import logging
+
+import torch.cuda
+
+logger = logging.getLogger(__name__)
+
+
+def test_cuda_availability() -> None:
+    """Test if CUDA is available."""
+    logger.info("Testing CUDA...")
+    # GIVEN / WHEN
+    cuda_available = torch.cuda.is_available()
+
+    # THEN
+    assert cuda_available is True
+
+    logger.info("CUDA available!")
+    logger.info("✓ CUDA version: %s", torch.version.cuda)
+    logger.info("✓ Number of GPUs: %d", torch.cuda.device_count())
+    for i in range(torch.cuda.device_count()):
+        logger.info("  - GPU %d: %s", i, torch.cuda.get_device_name(i))

gmtorch-0.0.0.dev0/tests/test_import.py

@@ -0,0 +1,8 @@
+"""Test gmtorch."""
+
+import gmtorch
+
+
+def test_import() -> None:
+    """Test that the package can be imported."""
+    assert isinstance(gmtorch.__name__, str)