Make the PyTorch backend work, and document JAX/PyTorch as selectable backends - #7

Open

geekbeast wants to merge 5 commits into

mainfrom

feature/pytorch-backend

Commits on Aug 26, 2026

Fix asymmetric diagonal sweep geometry and a jit tracer leak in the JAX backend
geekbeast
committed
a1ae5d6
Replace the PyTorch backend with a working implementation and cover it in CI
geekbeast
committed
2de91d9
Document JAX and PyTorch as selectable backends
geekbeast
committed
9b53296
Guard that the corrected sweep geometry stays free
geekbeast
committed
f56abe7
Stop compile_forward from aliasing the CUDA-graph output buffer
geekbeast
committed
b7857e4