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