Metadata-Version: 2.4
Name: transformer_engine_rocm_jax
Version: 2.14.0.dev0+rocm7.15.0a20260626.a4e9dd6
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.14.0.dev0+rocm7.15.0a20260626.a4e9dd6
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
