axlearn

PyPI page
Home page
Author:
License:
Summary: AXLearn
Latest version: 0.0.1.post20240213000026
Required dependencies: absl-py | attrs | chex | flax | importlab | jax | jaxlib | nltk | numpy | optax | portpicker | protobuf | seqio | tensorboard-plugin-profile | tensorflow | tensorflow-datasets | tensorflow-io | tensorflow_text | tensorstore | toml | typing-extensions
Optional dependencies: apache-beam | axlearn | black | cloud-tpu-client | crcmod | diffusers | einops | evaluate | fairseq | google-api-python-client | google-apitools | google-auth | google-cloud-aiplatform | google-cloud-core | google-cloud-storage | isort | jax | orjson | pre-commit | pycocotools | pylint | pyopenssl | pytest | pytest-xdist | pytype | scikit-learn | scipy | sentencepiece | setuptools | tensorboard | tensorflow-metal | timm | torch | torchvision | tqdm | transformers | wandb | wrapt

Downloads last day: 13
Downloads last week: 52
Downloads last month: 178