PyPI page
Home page
Author:
PAX team
License:
Apache-2.0
Summary:
Functionalities such as a layers for building neural networks in Jax.
Latest version:
1.4.0
Required dependencies:
absl-py
|
chex
|
clu
|
einops
|
etils
|
fiddle
|
flax
|
jax
|
jax-bitempered-loss
|
jaxtyping
|
lingvo
|
numpy
|
opt-einsum
|
optax
|
optax-shampoo
|
sentencepiece
|
tensorflow
|
tensorflow-datasets
|
tensorflow-metadata
|
tensorflow-text
|
tfds-nightly
|
typeguard
Downloads last day:
1,815
Downloads last week:
7,609
Downloads last month:
30,918