Skip to content

build: migrate to pyproject.toml + Nix flake with CPU/CUDA/ROCm dev shells#40

Open
mayrf wants to merge 2 commits into
mainfrom
polish/nix-flake
Open

build: migrate to pyproject.toml + Nix flake with CPU/CUDA/ROCm dev shells#40
mayrf wants to merge 2 commits into
mainfrom
polish/nix-flake

Conversation

@mayrf
Copy link
Copy Markdown
Owner

@mayrf mayrf commented May 19, 2026

Replaces setup.py/setup.cfg/versioneer with pyproject.toml (uv + setuptools-scm).

Adds a Nix flake with CPU, CUDA, and ROCm dev shells. direnv auto-detects the right
shell based on available hardware. torch is resolved from the CPU wheel index in all
Nix shells — GPU acceleration still works at runtime via LD_LIBRARY_PATH from Nix,
without downloading NVIDIA pip wheels.

Non-Nix users (pip, conda) are unaffected.

@mayrf mayrf force-pushed the polish/nix-flake branch from 8fb1dd6 to 6dd46ed Compare May 20, 2026 06:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant