hoptorch

What it is and what it does

hoptorch is a thin compatibility layer for PyTorch's scan operator, which is part of PyTorch's higher-order operator suite. It wraps torch._higher_order_ops.scan with a health check that verifies scan backward is working on your device before you use it, then lazily applies version-specific patches to PyTorch internals if needed. For PyTorch 2.7, where scan backward is not yet implemented, hoptorch provides a small eager scan implementation that uses ordinary autograd instead.

The package is designed to let you write scan code once and have it work reliably across PyTorch versions and devices. It includes utilities to check whether scan is available, to warm the health check before torch.compile, and to inspect why scan might be unavailable on a given device. The health check fails closed during Dynamo tracing to avoid false positives.

Use it for

• Use it to safely call torch.scan in production code without worrying about device-specific or version-specific backward failures.
• Use it to backport scan functionality to PyTorch 2.7 environments where native scan backward is not available.
• Use it to warm scan health checks before compiling functions with torch.compile to avoid tracing overhead.
• Use it to diagnose why scan backward is unavailable on a specific device or PyTorch version.

Worth the install?

AI-flagged interpretation of the facts on this page. Verify before relying on it.

Install

hoptorch on PyPI

Before you install

Low friction: pure Python wheel with only torch and pyvers as runtime dependencies. Active maintenance with recent release, though early-stage (Alpha) and minimal community adoption.

Requires torch>=2.7 and pyvers>=0.2.2; Python >=3.10

License in practice

MIT license is permissive; you can use, modify, and distribute this package with minimal restrictions.

Quickstart

pip install hoptorch

import torch
from hoptorch import scan
from hoptorch.scan import ensure_scan_backward

if ensure_scan_backward("cpu"):
    xs = torch.arange(4.0)
    def step(carry, x):
        next_carry = carry + x
        return next_carry, next_carry.clone()
    carry, ys = scan(step, torch.zeros(()), xs)

Package facts

Evidence: hoptorch-0.1.4-py3-none-any.whl

Tags

More Python Modules packages

See also torch · torch-einops-utils · cuequivariance-ops-torch-cu12 · torch-npu · pytorch · onnx2torch · torchviz · depyf · torch-directml · opt-einsum-fx