PyPI page
Home page
Author:
Erfan Zare Chavoshi
License:
Apache-2.0
Summary:
Accelerate, Optimize performance with streamlined training and serving options with JAX.
Latest version:
0.2.0.2
Required dependencies:
cryptography
|
datasets
|
eformer
|
einops
|
ejkernel
|
fastapi
|
flax
|
gcsfs
|
google-api-python-client
|
grafanalib
|
grain
|
jax
|
jaxlib
|
jinja2
|
msgspec
|
partial-json-parser
|
pyyaml
|
pyzmq
|
ray
|
sentencepiece
|
transformers
|
uvicorn
|
uvloop
|
wandb
|
zstandard
Optional dependencies:
ejkernel
|
jax
|
lm-eval
|
tb-nightly
|
tensorboard
|
tensorflow
|
tensorflow-datasets
|
torch
|
xprof
|
xprof-nightly
Downloads last day:
311
Downloads last week:
1,915
Downloads last month:
7,069