beamgrad 2.1.1

Differentiable beam search for PyTorch: exact beam search forward, surrogate gradients backward, native on CPU and CUDA.

hardwarereleases

Sources

beamgrad 2.1.1 · TechNews