Skip to content

v0.0.1.dev0

Pre-release
Pre-release

Choose a tag to compare

View all tags
@walkerchi walkerchi released this 22 Jun 16:20
v0.0.1.dev0
77acf64

First dev release of torch-strumpack — a differentiable sparse direct solver for PyTorch, backed by STRUMPACK.

Self-contained manylinux_2_39 wheels (Python 3.12, x86_64), real STRUMPACK (no scipy fallback), GPU offload verified:

  • +cpu — CPU, no GPU runtime needed
  • +rocm7.2.4 — AMD ROCm (built gfx1100; needs a ROCm install)
  • +cuda12.0 — NVIDIA CUDA (sm_89; needs a CUDA install)

CPU deps (OpenBLAS/METIS/libgomp) are bundled; GPU runtimes are provided by the user.

Assets 25
Loading