Metadata-Version: 2.4
Name: transformer_engine_rocm_jax
Version: 2.10.0
Summary: Transformer acceleration library - Jax Lib
Requires-Python: >=3.10.0
Requires-Dist: jax
Requires-Dist: flax>=0.7.1
Requires-Dist: transformer_engine_rocm7==2.10.0
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
