beamgrad 2.1.1
Differentiable beam search for PyTorch: exact beam search forward, surrogate gradients backward, native on CPU and CUDA.
Sources
- T1beamgrad 2.1.1PyPI / crates.io / RubyGems / Go index / NuGet
Differentiable beam search for PyTorch: exact beam search forward, surrogate gradients backward, native on CPU and CUDA.