Metadata-Version: 2.1
Name: transformer_engine_jax
Version: 1.9.0
Summary: Transformer acceleration library - Jax Lib
Requires-Dist: jax
Requires-Dist: flax>=0.7.1
