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:
38
Downloads last week:
1,248
Downloads last month:
8,161
7-day smoothing