Metadata-Version: 2.4
Name: transformer_engine_jax
Version: 2.6.0
Summary: Transformer acceleration library - Jax Lib
Requires-Dist: flax>=0.7.1
Requires-Dist: jax==0.6.2
Dynamic: requires-dist
Dynamic: summary
