PyPI page
Summary:
Transformer acceleration library - Jax Lib
Latest version:
2.20.2
Required dependencies:
flax
|
jax
|
nvidia-cudnn-frontend
|
transformer-engine-cu12
Downloads last day:
157
Downloads last week:
2,026
Downloads last month:
8,161
7-day smoothing