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