Metadata-Version: 2.1
Name: paxml
Version: 1.4.0
Summary: Framework to configure and run machine learning experiments on top of Jax.
Home-page: https://github.com/google/paxml
Author: PAX team
Author-email: pax-dev@google.com
License: Apache-2.0
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10
License-File: LICENSE
Requires-Dist: absl-py (==1.4.0)
Requires-Dist: clu (==0.0.11)
Requires-Dist: etils (==1.7.0)
Requires-Dist: flax (==0.8.2)
Requires-Dist: graphviz (==0.20.1)
Requires-Dist: jax (==0.4.26)
Requires-Dist: lingvo (==0.12.7)
Requires-Dist: numpy (==1.26.4)
Requires-Dist: orbax-checkpoint (==0.5.9)
Requires-Dist: praxis (==1.4.0)
Requires-Dist: protobuf (==3.19.6)
Requires-Dist: pyglove (==0.4.4)
Requires-Dist: seqio-nightly (==0.0.17.dev20231010)
Requires-Dist: t5 (==0.9.4)
Requires-Dist: tensorflow-datasets (==4.8.3)
Requires-Dist: tensorflow-metadata (==1.12.0)
Requires-Dist: tensorflow-text (~=2.9.0)
Requires-Dist: tensorflow (~=2.9.2)
Requires-Dist: tensorstore (==0.1.55)
Requires-Dist: tfds-nightly (==4.8.3.dev202303280045)
Provides-Extra: gpu
Requires-Dist: jsonlines (==3.1.0) ; extra == 'gpu'
Requires-Dist: pysimdjson (==5.0.2) ; extra == 'gpu'
Requires-Dist: zstandard (==0.18.0) ; extra == 'gpu'

UNKNOWN

