Metadata-Version: 2.4
Name: transformer_engine_rocm_jax
Version: 2.10.0+devrocm7.13.0.dev0.45223d0d4709b5ec27afa0a0efb53faf2d3502a3
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+devrocm7.13.0.dev0.45223d0d4709b5ec27afa0a0efb53faf2d3502a3
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
