PyPI page
Home page
Author:
PAX team
License:
Apache-2.0
Summary:
Framework to configure and run machine learning experiments on top of Jax.
Latest version:
1.4.0
Required dependencies:
absl-py
|
clu
|
etils
|
flax
|
graphviz
|
jax
|
lingvo
|
numpy
|
orbax-checkpoint
|
praxis
|
protobuf
|
pyglove
|
seqio-nightly
|
t5
|
tensorflow
|
tensorflow-datasets
|
tensorflow-metadata
|
tensorflow-text
|
tensorstore
|
tfds-nightly
Optional dependencies:
jsonlines
|
pysimdjson
|
zstandard
Downloads last day:
1,763
Downloads last week:
6,249
Downloads last month:
29,238