Metadata-Version: 2.1
Name: praxis
Version: 1.1.0
Summary: Functionalities such as a layers for building neural networks in Jax.
Home-page: https://github.com/google/praxis
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.9)
Requires-Dist: einops (==0.6.1)
Requires-Dist: etils[enp,epath,epy] (==1.4.1)
Requires-Dist: fiddle (==0.2.11)
Requires-Dist: flax (==0.7.2)
Requires-Dist: jax-bitempered-loss (==0.0.2)
Requires-Dist: jax (==0.4.14)
Requires-Dist: jaxtyping (==0.2.21)
Requires-Dist: lingvo (==0.12.7)
Requires-Dist: numpy (==1.25.2)
Requires-Dist: optax-shampoo (==0.0.6)
Requires-Dist: optax (==0.1.7)
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: tfds-nightly (==4.8.3.dev202303280045)
Requires-Dist: typeguard (==4.1.2)

UNKNOWN

