Warprnnt-numba

Latest version: v0.4.1

Safety actively analyzes 626118 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

0.4.0

Supports
- Simple RNNT loss with Atomic Locks implementation

Improvements
- Improve runtime speed of numba loss
- Fix issue with data movement of costs tensor from llForward to pytorch data view in numba
- This alone costs a linear loop (scaling with batch size) that is roughly 10x the kernel costs themselves.
- Fix by writing a small kernel to copy the data and update the costs.

0.2.2

Initial release of Warp RNNT loss with Numba JIT compile (CPU/CUDA)

Supports:
1) Pytorch RNNT loss (CPU and JIT compiled CUDA)
2) FastEmit
3) Gradient clipping

Links

Releases

Has known vulnerabilities

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.